| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test_common.h | 91 this->BuildInterpreter({this->GetShape(input_)}); in BuildWithHardcodedShape()
107 this->BuildInterpreter( in BuildWithTensorShape()
127 this->BuildInterpreter({this->GetShape(input_)}); in BuildWithConstantTensorShape()
|
| D | complex_support_test.cc | 40 BuildInterpreter({GetShape(input_)}); in RealOpModel()
101 BuildInterpreter({GetShape(input_)}); in ImagOpModel()
162 BuildInterpreter({GetShape(input_)}); in ComplexAbsOpModel()
|
| D | unsorted_segment_test.h | 43 BuildInterpreter({GetShape(data_id_), GetShape(segment_ids_id_), in UnsortedSegmentModel()
60 BuildInterpreter({GetShape(data_id_), GetShape(segment_ids_id_), in UnsortedSegmentModel()
|
| D | lsh_projection_test.cc | 47 BuildInterpreter({hash_shape, input_shape, weight_shape}); in LSHProjectionOpModel()
49 BuildInterpreter({hash_shape, input_shape}); in LSHProjectionOpModel()
|
| D | reduce_test_common.h | 86 BuildInterpreter({GetShape(input_)}); in BaseConstOpModel()
105 BuildInterpreter({GetShape(input_)}); in BaseDynamicOpModel()
|
| D | random_uniform_test.cc | 66 BuildInterpreter({GetShape(input_)}); in RandomUniformOpModel()
95 BuildInterpreter({GetShape(input_)}); in RandomUniformIntOpModel()
|
| D | dequantize_test.cc | 61 BuildInterpreter({GetShape(input_)}); in DequantizeOpModel()
141 BuildInterpreter({GetShape(input_)}); in DequantizePerChannelOpModel()
|
| D | pad_test.cc | 90 this->BuildInterpreter({input.shape}); in PadV2OpConstModel()
107 this->BuildInterpreter({input.shape}); in PadV2OpConstModel()
130 BuildInterpreter({input.shape}); in PadOpConstModel()
150 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
163 this->BuildInterpreter({input.shape, paddings_shape}); in PadV2OpDynamicModel()
186 this->BuildInterpreter({input.shape, paddings_shape}); in PadOpDynamicModel()
|
| D | split_test.cc | 55 BuildInterpreter({GetShape(axis_), GetShape(input_)}); in SplitOpModel()
57 BuildInterpreter({{}, GetShape(input_)}); in SplitOpModel()
|
| D | tile_test.cc | 67 BuildInterpreter({input_shape, {static_cast<int>(input_shape.size())}}); in TileOpConstModel()
79 BuildInterpreter({input_shape, {static_cast<int>(input_shape.size())}}); in TileOpDynamicModel()
|
| D | split_v_test.cc | 61 BuildInterpreter( in SplitVOpModel()
64 BuildInterpreter({GetShape(input_), GetShape(size_splits_), {}}); in SplitVOpModel()
|
| D | floor_mod_test_common.h | 31 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in FloorModModel()
|
| D | random_uniform_custom_test.cc | 68 BuildInterpreter({GetShape(input_)}); in RandomUniformOpModel()
98 BuildInterpreter({GetShape(input_)}); in RandomUniformIntOpModel()
|
| D | batch_to_space_nd_test.cc | 83 BuildInterpreter({input_shape}); in BatchToSpaceNDOpConstModel()
108 BuildInterpreter({input_shape, {spatial_dims}, {spatial_dims, 2}}); in BatchToSpaceNDOpDynamicModel()
|
| D | quant_basic_lstm_test.cc | 114 BuildInterpreter({GetShape(input_), GetShape(zero_input_), in QuantizedLSTMOpModel()
118 BuildInterpreter({GetShape(input_), GetShape(prev_output_), in QuantizedLSTMOpModel()
|
| D | where_test.cc | 40 BuildInterpreter({GetShape(input_)}); in BaseWhereOpModel()
65 BuildInterpreter({{}}); in ConstInputWhereOpModel()
|
| D | non_max_suppression_test.cc | 92 BuildInterpreter({GetShape(input_boxes_), GetShape(input_scores_), in NonMaxSuppressionV4OpModel()
185 BuildInterpreter( in NonMaxSuppressionV5OpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/ |
| D | opencl_test.cc | 42 BuildInterpreter(GetModel(model_binary.data()), &tflite_cpu_interpreter)); in TEST_P()
46 BuildInterpreter(GetModel(model_binary.data()), &opencl_interpreter)); in TEST_P()
|
| D | opengl_test.cc | 40 BuildInterpreter(GetModel(model_binary.data()), &tflite_cpu_interpreter)); in TEST_P()
44 BuildInterpreter(GetModel(model_binary.data()), &opengl_interpreter)); in TEST_P()
|
| D | xnnpack_test.cc | 41 BuildInterpreter(GetModel(model_binary.data()), &tflite_cpu_interpreter)); in TEST_P()
45 BuildInterpreter(GetModel(model_binary.data()), &xnnpack_interpreter)); in TEST_P()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.h | 43 BuildInterpreter({GetShape(data_id_), GetShape(segment_ids_id_), in UnsortedSegmentModel()
60 BuildInterpreter({GetShape(data_id_), GetShape(segment_ids_id_), in UnsortedSegmentModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | arg_min_max_test.cc | 54 BuildInterpreter({input_shape_, {1}}); in Build()
70 BuildInterpreter({input_shape_, {1}}); in Build()
|
| D | min_max_builder_test.cc | 34 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in MinMaxOpModel()
47 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in MinMaxOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/gradient/ |
| D | bcast_grad_args_test.cc | 46 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in BcastGradArgsInt32OpModel()
148 BuildInterpreter({GetShape(input1_), GetShape(input2_)}); in BcastGradArgsInt64OpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_device_selection_test.cc | 58 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, /*num_threads=*/-1, in Init()
278 BuildInterpreter({input_shape, {1}}); in Init()
416 BuildInterpreter({GetShape(input1_), GetShape(input2_), GetShape(input3_)}, in Init()
598 BuildInterpreter({GetShape(input1_), GetShape(input2_)}, /*num_threads=*/-1, in Init()
727 BuildInterpreter({GetShape(input_), GetShape(filter_), GetShape(bias_)},
875 BuildInterpreter({GetShape(input_), GetShape(zero_input_)}); in Init()
|