Home
last modified time | relevance | path

Searched refs:input_tensor_id (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad_test.cc41 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::__anon97c8e8c00111::BaseMirrorPadOpModel
56 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
66 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
77 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
88 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
99 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
110 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
122 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
134 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
148 model.PopulateTensor<int>(model.input_tensor_id(), {1, 2, 3, 4, 5, 6}); in TEST()
[all …]
Dunique_test.cc39 int input_tensor_id() { return input_id_; } in input_tensor_id() function in tflite::__anon92b047880111::UniqueOpModel
55 model.PopulateTensor<float>(model.input_tensor_id(), {5}); in TEST()
64 model.PopulateTensor<float>(model.input_tensor_id(), in TEST()
75 model.PopulateTensor<float>(model.input_tensor_id(), {5, 5, 5, 5, 5, 5, 5}); in TEST()
85 model.PopulateTensor<float>(model.input_tensor_id(), {2, 3, 5, 7, 2, 7, 3}); in TEST()
95 model.PopulateTensor<float>(model.input_tensor_id(), {2, 3, 5, 7, 2, 7, 3}); in TEST()
/external/tensorflow/tensorflow/c/eager/
Dtape.h37 std::vector<int64> input_tensor_id; member
131 gtl::ArraySlice<int64> input_tensor_id,
208 gtl::ArraySlice<int64> input_tensor_id, in RecordOperation() argument
212 if (!ShouldRecord(input_tensor_id, input_dtypes)) { in RecordOperation()
216 ids.reserve(input_tensor_id.size()); in RecordOperation()
217 for (int64 i : input_tensor_id) { in RecordOperation()
266 for (int64 id : op_it->second.input_tensor_id) { in DeleteTrace()
348 for (auto it : op_it->second.input_tensor_id) { in PrepareBackprop()
570 in_gradients.resize(trace.input_tensor_id.size()); in ComputeGradient()
581 << trace.input_tensor_id.size() << " sources"; in ComputeGradient()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc96 TensorId input_tensor_id = ParseTensorName(input_name); in ExtractExtraProperties() local
97 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties()
169 TensorId input_tensor_id = ParseTensorName(input_name); in FindInputFeatures() local
170 const string input_node_name(input_tensor_id.first); in FindInputFeatures()
171 const int output_index = input_tensor_id.second; in FindInputFeatures()