Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier_internal_test.cc61 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
H A Dverifier_test.cc83 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
/aosp_15_r20/external/pytorch/test/mobile/lightweight_dispatch/
H A Dtest_codegen_unboxing.cpp160 TEST(LiteInterpreterTest, AddTensor) { in TEST() argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset.h224 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc1713 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder