Searched defs:AddTensor (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
H A D | subgraph.cc | 27 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 D | session_state.cc | 37 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 D | verifier_internal_test.cc | 61 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
|
H A D | verifier_test.cc | 83 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 D | test_codegen_unboxing.cpp | 160 TEST(LiteInterpreterTest, AddTensor) { in TEST() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | dataset.h | 224 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | nnapi_delegate.cc | 1713 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder
|