Home
last modified time | relevance | path

Searched defs:AddTensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsubgraph.cc27 int GPUOperationsSubgraph::AddTensor(const TensorDescriptor& desc) { in AddTensor() function in tflite::gpu::GPUOperationsSubgraph
32 int GPUOperationsSubgraph::AddTensor(const BHWC& shape, in AddTensor() function in tflite::gpu::GPUOperationsSubgraph
39 int GPUOperationsSubgraph::AddTensor(const OHWI& shape, in AddTensor() function in tflite::gpu::GPUOperationsSubgraph
/external/tensorflow/tensorflow/core/common_runtime/
Dsession_state.cc37 Status SessionState::AddTensor(const string& handle, const Tensor& tensor) { in AddTensor() function in tensorflow::SessionState
60 Status TensorStore::AddTensor(const string& name, const TensorAndKey& tk) { in AddTensor() function in tensorflow::TensorStore
/external/tensorflow/tensorflow/lite/tools/
Dverifier_internal_test.cc61 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
Dverifier_test.cc83 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
/external/pytorch/test/mobile/lightweight_dispatch/
Dtest_codegen_unboxing.cpp160 TEST(LiteInterpreterTest, AddTensor) { in TEST() argument
/external/tensorflow/tensorflow/core/framework/
Ddataset.h224 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc1713 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder