HomeSort by relevance Sort by last modified time
    Searched refs:TensorData (Results 1 - 16 of 16) sorted by null

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

Completed in 332 milliseconds