frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Automatic generation of unittest for Glow integration

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19936

Reviewed By: ipiszy

Differential Revision: D15138090

fbshipit-source-id: 29a812548bb5da00176b00c1e9a26a7c31cea9c0
This commit is contained in:
Yinghai Lu
2019-04-30 12:03:42 -07:00
committed by Facebook Github Bot
parent 3a0727e58b
commit 9a81d1e692
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/predictor/emulator/data_filler.cc
View File

@ -22,7 +22,7 @@ void DataNetFiller::fill_input_internal(TensorList_t* input_data) const {
}
}
static void fill_with_type(
void fill_with_type(
const TensorFiller& filler,
const std::string& type,
TensorCPU* output) {

13
caffe2/predictor/emulator/data_filler.h
View File

@ -79,6 +79,11 @@ class DataNetFiller : public Filler {
const NetDef data_net_;
};
void fill_with_type(
const TensorFiller& filler,
const std::string& type,
TensorCPU* output);
/*
* @run_net: the predict net with parameter and input names
* @input_dims: the input dimentions of all operator inputs of run_net
@ -95,10 +100,7 @@ class DataRandomFiller : public Filler {
void fill_parameter(Workspace* ws) const override;
protected:
DataRandomFiller() {}
TensorFiller get_tensor_filler(
static TensorFiller get_tensor_filler(
const OperatorDef& op_def,
int input_index,
const std::vector<std::vector<int64_t>>& input_dims) {
@ -117,6 +119,9 @@ class DataRandomFiller : public Filler {
return filler;
}
protected:
DataRandomFiller() {}
using filler_type_pair_t = std::pair<TensorFiller, std::string>;
std::unordered_map<std::string, filler_type_pair_t> parameters_;
std::unordered_map<std::string, filler_type_pair_t> inputs_;