| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 100 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, in FloatAddOpModel()
171 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in FloatMulOpModel()
210 BuildInterpreter({GetShape(input_)}); in FloatPoolingOpModel()
302 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in ConvolutionOpModel()
479 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)}); in DepthwiseConvolutionOpModel()
552 BuildInterpreter({GetShape(input_), GetShape(weights_), GetShape(bias_)}); in FullyConnectedOpModel()
632 BuildInterpreter({{batches_, input_size_}}); in SoftmaxOpModel()
683 BuildInterpreter({input_shape, {static_cast<int>(new_shape.size())}}); in ReshapeOpModel()
716 BuildInterpreter({GetShape(input_)}); in SqueezeOpModel()
771 BuildInterpreter({GetShape(input_)}); in L2NormOpModel()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test.cc | 88 BuildInterpreter({GetShape(input_)}); in BuildWithHardcodedShape()
102 BuildInterpreter({GetShape(input_), GetShape(shape_input_tensor)}); in BuildWithTensorShape()
119 BuildInterpreter({GetShape(input_)}); in BuildWithConstantTensorShape()
|
| D | lsh_projection_test.cc | 46 BuildInterpreter({hash_shape, input_shape, weight_shape}); in LSHProjectionOpModel()
48 BuildInterpreter({hash_shape, input_shape}); in LSHProjectionOpModel()
|
| D | reduce_test.cc | 68 BuildInterpreter({GetShape(input_)}); in MeanOpConstModel()
82 BuildInterpreter({GetShape(input_)}); in MeanOpDynamicModel()
97 BuildInterpreter({GetShape(input_)}); in SumOpConstModel()
111 BuildInterpreter({GetShape(input_)}); in SumOpDynamicModel()
126 BuildInterpreter({GetShape(input_)}); in ProdOpConstModel()
140 BuildInterpreter({GetShape(input_)}); in ProdOpDynamicModel()
155 BuildInterpreter({GetShape(input_)}); in MaxOpConstModel()
169 BuildInterpreter({GetShape(input_)}); in MaxOpDynamicModel()
184 BuildInterpreter({GetShape(input_)}); in MinOpConstModel()
198 BuildInterpreter({GetShape(input_)}); in MinOpDynamicModel()
[all …]
|
| D | pad_test.cc | 86 this->BuildInterpreter({input.shape}); in PadV2OpConstModel()
103 this->BuildInterpreter({input.shape}); in PadV2OpConstModel()
126 BuildInterpreter({input.shape}); in PadOpConstModel()
146 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
159 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
182 this->BuildInterpreter({input.shape, paddings_shape}); in PadOpDynamicModel()
|
| D | batch_to_space_nd_test.cc | 75 BuildInterpreter({input_shape}); in BatchToSpaceNDOpConstModel()
99 BuildInterpreter({input_shape, {2}, {2, 2}}); in BatchToSpaceNDOpDynamicModel()
|
| D | split_test.cc | 44 BuildInterpreter({GetShape(axis_), GetShape(input_)}); in SplitOpModel()
46 BuildInterpreter({{}, GetShape(input_)}); in SplitOpModel()
|
| D | elementwise_test.cc | 44 BuildInterpreter({input_shape}); in ElementWiseOpFloatModel()
55 BuildInterpreter({input_shape}); in ElementWiseOpBoolModel()
|
| D | activations_test.cc | 41 BuildInterpreter({GetShape(input_)}); in BaseActivationsOpModel()
56 BuildInterpreter({GetShape(input_)}); in BaseActivationsOpModel()
64 BuildInterpreter({GetShape(input_)}); in BaseActivationsOpModel()
844 BuildInterpreter({GetShape(input_), GetShape(alpha_)}); in BasePReluOpModel()
948 BuildInterpreter({GetShape(input_)}); in LeakyReluOpModel()
|
| D | split_v_test.cc | 46 BuildInterpreter( in SplitVOpModel()
49 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()
|
| D | concatenation_test.cc | 45 BuildInterpreter(all_input_shapes); in BaseConcatenationOpModel()
79 BuildInterpreter(all_input_shapes); in QuantizedConcatenationOpModel()
|
| D | arg_min_max_test.cc | 59 ArgBaseOpModel<T>::BuildInterpreter({input_shape, {1, 1, 1, 1}}); in ArgMaxOpModel()
74 ArgBaseOpModel<T>::BuildInterpreter({input_shape, {1, 1, 1, 1}}); in ArgMinOpModel()
|
| D | exp_test.cc | 33 BuildInterpreter({GetShape(input_)}); in ExpOpModel()
|
| D | ceil_test.cc | 33 BuildInterpreter({ in CeilOpModel()
|
| D | floor_test.cc | 33 BuildInterpreter({ in FloorOpModel()
|
| D | dequantize_test.cc | 39 BuildInterpreter({GetShape(input_)}); in DequantizeOpModel()
|
| D | space_to_batch_nd_test.cc | 83 BuildInterpreter({input.shape}); in SpaceToBatchNDOpConstModel()
107 BuildInterpreter({input.shape, {2}, {2, 2}}); in SpaceToBatchNDOpDynamicModel()
|
| D | local_response_norm_test.cc | 37 BuildInterpreter({input_shape}); in LocalResponseNormOpModel()
|
| D | neg_test.cc | 33 BuildInterpreter({GetShape(input_)}); in NegOpModel()
|
| D | zeros_like_test.cc | 33 BuildInterpreter({GetShape(input_)}); in ZerosLikeOpModel()
|
| D | rank_test.cc | 37 BuildInterpreter({input_shape}); in RankOpModel()
|
| D | mfcc_test.cc | 56 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in BaseMfccOpModel()
|
| D | expand_dims_test.cc | 37 BuildInterpreter({input_shape, {1}}); in ExpandDimsOpModel()
|
| /external/libtextclassifier/utils/tflite/ |
| D | dist_diversification_test.cc | 64 BuildInterpreter({{matrix_rows, matrix_rows}, {1}, {1}}); in DistanceDiversificationOpModel()
|
| /external/tensorflow/tensorflow/lite/models/smartreply/ops/ |
| D | normalize_test.cc | 42 BuildInterpreter({{static_cast<int>(input.size())}}); in NormalizeOpModel()
|