| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad_test.cc | 42 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::__anonfefff6de0111::BaseMirrorPadOpModel
57 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
67 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
78 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
89 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
100 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
111 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
123 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
135 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
149 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
[all …]
|
| D | unique_test.cc | 42 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::__anon2d08f6a60111::UniqueOpModel
58 model.PopulateTensor<float>(model.input_tensor_id(), {5}); in TEST()
67 model.PopulateTensor<float>(model.input_tensor_id(), in TEST()
78 model.PopulateTensor<float>(model.input_tensor_id(), {5, 5, 5, 5, 5, 5, 5}); in TEST()
88 model.PopulateTensor<float>(model.input_tensor_id(), {2, 3, 5, 7, 2, 7, 3}); in TEST()
98 model.PopulateTensor<float>(model.input_tensor_id(), in TEST()
108 model.PopulateTensor<float>(model.input_tensor_id(), {2, 3, 5, 7, 2, 7, 3}); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | mirror_pad_test.cc | 37 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::MirrorPadOpModel
51 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
60 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
71 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
82 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
92 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
102 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3}); in TEST()
111 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3}); in TEST()
120 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | arg_min_max_builder.cc | 34 int input_tensor_id = inputs->data[0]; in PopulateSubGraph() local
35 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph()
36 AddInput(graph_builder_->GetHexagonTensorId(input_tensor_id)); in PopulateSubGraph()
|
| D | split_builder.cc | 31 const int input_tensor_id = inputs->data[1]; in PopulateSubGraph() local
32 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph()
55 AddInput(graph_builder_->GetHexagonTensorId(input_tensor_id)); in PopulateSubGraph()
|
| D | resize_bilinear_builder.cc | 31 int input_tensor_id = inputs->data[0]; in PopulateSubGraph() local
32 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph()
33 AddInput(graph_builder_->GetHexagonTensorId(input_tensor_id)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 43 std::vector<int64> input_tensor_id; member
159 gtl::ArraySlice<int64> input_tensor_id,
284 gtl::ArraySlice<int64> input_tensor_id,
408 gtl::ArraySlice<int64> input_tensor_id, in RecordOperation() argument
412 if (!ShouldRecord(input_tensor_id, input_dtypes)) { in RecordOperation()
416 ids.reserve(input_tensor_id.size()); in RecordOperation()
417 for (int64 i : input_tensor_id) { in RecordOperation()
466 for (int64 id : op_it->second.input_tensor_id) { in DeleteTrace()
548 for (auto it : op_it->second.input_tensor_id) { in PrepareBackprop()
735 for (int i = 0, end = trace.input_tensor_id.size(); i < end; i++) { in ComputeGradient()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 97 TensorId input_tensor_id = ParseTensorName(input_name); in ExtractExtraProperties() local
98 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties()
170 TensorId input_tensor_id = ParseTensorName(input_name); in FindInputFeatures() local
171 const string input_node_name(input_tensor_id.first); in FindInputFeatures()
172 const int output_index = input_tensor_id.second; in FindInputFeatures()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3077 const int input_tensor_id = in Map() local
3087 input_tensor_id, hybrid_op, in Map()
3092 input_tensor_id, hybrid_op, in Map()
|