Searched refs:TensorData (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | mean_test.cc | 51 MeanOpConstModel(const TensorData& input, const TensorData& output, in MeanOpConstModel()
66 MeanOpDynamicModel(const TensorData& input, const TensorData& output, in MeanOpDynamicModel()
67 const TensorData& axis, bool keep_dims) { in MeanOpDynamicModel()
|
| D | test_util.h | 79 struct TensorData { struct
115 int AddInput(TensorType type) { return AddInput(TensorData{type}); } in AddInput()
116 int AddInput(const TensorData& t);
126 int AddOutput(TensorType type) { return AddOutput(TensorData{type}); } in AddOutput()
127 int AddOutput(const TensorData& t);
210 int AddTensor(TensorData t, std::initializer_list<int> data);
212 std::map<int, TensorData> tensor_data_;
|
| D | depthwise_conv_test.cc | 31 BaseDepthwiseConvolutionOpModel(const TensorData& input, in BaseDepthwiseConvolutionOpModel()
32 const TensorData& filter, in BaseDepthwiseConvolutionOpModel()
33 const TensorData& output) { in BaseDepthwiseConvolutionOpModel()
45 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseDepthwiseConvolutionOpModel()
|
| D | mul_test.cc | 28 BaseMulOpModel(const TensorData& input1, const TensorData& input2, in BaseMulOpModel()
29 const TensorData& output, in BaseMulOpModel()
|
| D | conv_test.cc | 46 TfLiteRegistration* registration, const TensorData& input, in BaseConvolutionOpModel()
47 const TensorData& filter, const TensorData& output, int stride_width = 2, in BaseConvolutionOpModel()
61 TensorData bias{TensorType_INT32, {bias_size}, 0, 0, bias_scale}; in BaseConvolutionOpModel()
|
| D | test_util.cc | 52 int SingleOpModel::AddTensor(TensorData t, std::initializer_list<int> data) { in AddTensor()
105 int SingleOpModel::AddInput(const TensorData& t) { in AddInput()
114 int id = AddTensor(TensorData{type, shape}, data); in AddConstInput()
125 int SingleOpModel::AddOutput(const TensorData& t) { in AddOutput()
|
| D | fully_connected_test.cc | 122 BaseFullyConnectedOpModel(int units, int batches, const TensorData& input, in BaseFullyConnectedOpModel()
123 const TensorData& output = {TensorType_FLOAT32}) in BaseFullyConnectedOpModel()
142 TensorData bias{TensorType_INT32, {units_}, 0, 0, bias_scale};
|
| D | squeeze_test.cc | 29 BaseSqueezeOpModel(const TensorData& input, const TensorData& output, in BaseSqueezeOpModel()
|
| D | add_test.cc | 28 BaseAddOpModel(const TensorData& input1, const TensorData& input2, in BaseAddOpModel()
29 const TensorData& output, in BaseAddOpModel()
|
| D | pooling_test.cc | 31 BasePoolingOpModel(BuiltinOperator type, const TensorData& input, in BasePoolingOpModel()
33 const TensorData& output) { in BasePoolingOpModel()
|
| D | activations_test.cc | 31 BaseActivationsOpModel(BuiltinOperator type, TensorData input) { in BaseActivationsOpModel()
43 BaseActivationsOpModel(float softmax_beta, TensorData input) { in BaseActivationsOpModel()
|
| D | exp_test.cc | 28 ExpOpModel(const TensorData& input, const TensorType& output) { in ExpOpModel()
|
| D | space_to_depth_test.cc | 29 SpaceToDepthOpModel(const TensorData& tensor_data, int block_size) { in SpaceToDepthOpModel()
|
| D | split_test.cc | 30 SplitOpModel(const TensorData& input, int num_splits, in SplitOpModel()
|
| D | resize_bilinear_test.cc | 28 ResizeBilinearOpModel(const TensorData& input, in ResizeBilinearOpModel()
|
| D | concatenation_test.cc | 31 BaseConcatenationOpModel(const TensorData& input_template, int axis, in BaseConcatenationOpModel()
|