Searched refs:input_template (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation_test.cc | 32 BaseConcatenationOpModel(const TensorData& input_template, int axis, in BaseConcatenationOpModel() argument
36 all_input_shapes.push_back(input_template.shape); in BaseConcatenationOpModel()
37 AddInput(input_template); in BaseConcatenationOpModel()
39 output_ = AddOutput({input_template.type, /*shape=*/{}, input_template.min, in BaseConcatenationOpModel()
40 input_template.max}); in BaseConcatenationOpModel()
64 QuantizedConcatenationOpModel(const std::vector<TensorData>& input_template, in QuantizedConcatenationOpModel() argument
68 CHECK_EQ(input_template.size(), num_inputs); in QuantizedConcatenationOpModel()
70 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
71 AddInput(input_template[i]); in QuantizedConcatenationOpModel()
|
| D | pack_test.cc | 30 PackOpModel(const TensorData& input_template, int axis, int values_count) { in PackOpModel() argument
33 all_input_shapes.push_back(input_template.shape); in PackOpModel()
34 AddInput(input_template); in PackOpModel()
36 output_ = AddOutput({input_template.type, /*shape=*/{}, input_template.min, in PackOpModel()
37 input_template.max}); in PackOpModel()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/tools/make/ |
| D | generate_keil_project_test.sh | 26 --input_template=${INPUT_TEMPLATE} \
|
| D | generate_keil_project.py | 33 with open(flags.input_template, 'r') as input_template_file:
|
| D | helper_functions.inc | 78 --input_template=$$< --output_file=$$@ --executable=$(3) \
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 907 BaseConcatenationOpModel(const TensorData& input_template, int axis, in BaseConcatenationOpModel() argument
911 all_input_shapes.push_back(input_template.shape); in BaseConcatenationOpModel()
912 AddInput(input_template); in BaseConcatenationOpModel()
914 output_ = AddOutput({input_template.type, /*shape=*/{}, input_template.min, in BaseConcatenationOpModel()
915 input_template.max}); in BaseConcatenationOpModel()
962 QuantizedConcatenationOpModel(const std::vector<TensorData>& input_template, in QuantizedConcatenationOpModel() argument
966 CHECK_EQ(input_template.size(), num_inputs); in QuantizedConcatenationOpModel()
968 all_input_shapes.push_back(input_template[i].shape); in QuantizedConcatenationOpModel()
969 AddInput(input_template[i]); in QuantizedConcatenationOpModel()
|