| /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::__anon0c082fed0111::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::__anon051900350111::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 | 39 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::MirrorPadOpModel
53 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
62 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
73 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
84 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
94 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
104 model.PopulateTensor<int8_t>(model.input_tensor_id(), {1, 2, 3}); in TEST()
113 model.PopulateTensor<uint8_t>(model.input_tensor_id(), {1, 2, 3}); in TEST()
122 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()
54 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_t> input_tensor_id; member
159 gtl::ArraySlice<int64_t> input_tensor_id,
284 gtl::ArraySlice<int64_t> input_tensor_id,
408 gtl::ArraySlice<int64_t> 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_t i : input_tensor_id) { in RecordOperation()
466 for (int64_t 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 | 98 TensorId input_tensor_id = ParseTensorName(input_name); in ExtractExtraProperties() local
99 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties()
171 TensorId input_tensor_id = ParseTensorName(input_name); in FindInputFeatures() local
172 const string input_node_name(input_tensor_id.first); in FindInputFeatures()
173 const int output_index = input_tensor_id.second; in FindInputFeatures()
|
| /external/armnn/python/pyarmnn/test/ |
| D | test_runtime.py | 27 input_tensor_id = input_binding_info[0]
36 const_tensor_pair = (input_tensor_id, ann.ConstTensor(input_tensor_info, input_data))
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 3700 const int input_tensor_id = in Map() local
3706 add_zero_bias(input_tensor_id, filter_tensor_id, num_units); in Map()
3945 const int input_tensor_id = in Map() local
3955 input_tensor_id, hybrid_op, in Map()
3960 input_tensor_id, hybrid_op, in Map()
3984 add_zero_bias(input_tensor_id, weight_tensor_id, output_depth); in Map()
|