| /external/tensorflow/tensorflow/cc/framework/ |
| D | testutil.cc | 27 void GetTensors(const Scope& scope, OutputList tensors, in GetTensors() function
35 GetTensors(scope, {std::move(tensor)}, &outputs); in GetTensor()
39 void GetTensors(const Scope& scope, const std::vector<Output>& assign_vars, in GetTensors() function
49 GetTensors(scope, assign_vars, {std::move(tensor)}, &outputs); in GetTensor()
|
| D | testutil.h | 26 void GetTensors(const Scope& scope, OutputList tensors,
32 void GetTensors(const Scope& scope, const std::vector<Output>& assign_vars,
|
| D | cc_ops_test.cc | 91 test::GetTensors(root, {u.y, u.idx}, &outputs); in TEST()
141 test::GetTensors(root, {add}, &out); in TEST()
149 test::GetTensors(root, {sub, is_inited}, &out); in TEST()
|
| D | gradients_test.cc | 334 test::GetTensors(scope_test_, {grad_outputs[0], grad_outputs[1]}, &outputs); in TEST_F()
363 test::GetTensors(scope_test_, in TEST_F()
400 test::GetTensors(scope_test_, {x_assign, y_assign, z_assign}, in TEST_F()
433 test::GetTensors(scope_test_, {x_assign, y_assign, z_assign}, in TEST_F()
|
| /external/armnn/src/backends/tosaCommon/test/ |
| D | TosaTestUtils.hpp | 200 CHECK(basicBlock->GetTensors().size() == (numInputs + numOutputs)); in AssertTosaOneToOneMappingBasicBlock()
210 std::basic_string<char> tensorName = basicBlock->GetTensors()[i]->GetName(); in AssertTosaOneToOneMappingBasicBlock()
223 std::basic_string<char> tensorName = basicBlock->GetTensors()[numInputs + i]->GetName(); in AssertTosaOneToOneMappingBasicBlock()
241 TosaSerializationTensor* tensor = basicBlock->GetTensors()[i]; in AssertTosaOneToOneMappingBasicBlock()
249 TosaSerializationTensor* tensor = basicBlock->GetTensors()[i + inputShape.size()]; in AssertTosaOneToOneMappingBasicBlock()
|
| D | AvgPool2DIgnoreValueChecker.hpp | 26 CHECK(basicBlock->GetTensors().size() == 3); in VerifyAvgPool2DIgnoreValue()
|
| D | OneToOneMappingTests.cpp | 645 CHECK(basicBlock->GetTensors().size() == 4);
707 CHECK(basicBlock->GetTensors().size() == 4);
785 CHECK(basicBlock->GetTensors().size() == 0);
|
| /external/armnn/src/backends/tosaReference/ |
| D | TosaRefLayerSupport.cpp | 109 mappings->GetTensors(), in IsLayerSupported()
|
| D | TosaRefBackend.cpp | 136 auto blockTensors = mappings->GetTensors(); in OptimizeSubgraphView()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 158 tensorflow::gtl::InlinedVector<tensorflow::Tensor, 2>* GetTensors() { in GetTensors() function in tflite::flex::OpOutputs
348 auto* tensors = outputs_.GetTensors(); in MaybePersistTensorflowOutputs()
431 auto& outputs = *node_data->mutable_outputs()->GetTensors(); in ExecuteOpKernelRunner()
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | lazy_graph_executor.h | 89 std::vector<at::Tensor> GetTensors(std::vector<LazyTensorPtr>* tensors);
|
| D | lazy_graph_executor.cpp | 455 std::vector<at::Tensor> LazyGraphExecutor::GetTensors( in GetTensors() function in torch::lazy::LazyGraphExecutor
|