Home
last modified time | relevance | path

Searched refs:AddTensor (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier_test.cc83 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
184 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
185 builder.AddTensor({}, TensorType_UINT8, {}, "empty_vector"); in TEST()
186 builder.AddTensor( in TEST()
190 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
199 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
200 builder.AddTensor( in TEST()
204 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
236 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
237 builder.AddTensor( in TEST()
[all …]
H A Dverifier_internal_test.cc61 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
152 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
153 builder.AddTensor( in TEST()
157 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
/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
36 return AddTensor(desc_with_shape); in AddTensor()
39 int GPUOperationsSubgraph::AddTensor(const OHWI& shape, in AddTensor() function in tflite::gpu::GPUOperationsSubgraph
42 return AddTensor(shape_as_bhwc, desc); in AddTensor()
H A Dsubgraph.h45 int AddTensor(const TensorDescriptor& desc);
46 int AddTensor(const BHWC& shape, const TensorDescriptor& desc);
47 int AddTensor(const OHWI& shape, const TensorDescriptor& desc);
H A Doperation_selector.cc106 gpu_subgraph->AddTensor(src_transformed_shape, src_transformed_desc); in WinogradFromNode()
108 gpu_subgraph->AddTensor(dst_transformed_shape, dst_transformed_desc); in WinogradFromNode()
208 int tensor_id = gpu_subgraph->AddTensor( in AddDynamicConv()
217 int tensor_id = gpu_subgraph->AddTensor( in AddDynamicConv()
285 int tensor_id = gpu_subgraph->AddTensor(std::move(weights_tensor)); in AddConvSharedWeights()
301 int tensor_id = gpu_subgraph->AddTensor(std::move(weights_tensor)); in AddConvSharedWeights()
345 src_id = gpu_subgraph->AddTensor(hwc_input_shape, hwc_input_desc); in GPUOperationFromNodePart0()
346 dst_id = gpu_subgraph->AddTensor(hwc_output_shape, hwc_output_desc); in GPUOperationFromNodePart0()
430 int tensor_id = gpu_subgraph->AddTensor(concatenated_shape, in GPUOperationFromNodePart0()
/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
82 TF_RETURN_IF_ERROR(session_state->AddTensor(key, it->second.tensor)); in SaveTensors()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dsession_state.h37 Status AddTensor(const std::string& handle, const Tensor& tensor);
71 Status AddTensor(const std::string& name, const TensorAndKey& tk);
H A Ddataset.h224 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
H A Ddataset.cc803 return AddTensor(t, output); in AddDatasetOrTensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtest_util.cc89 id = AddTensor<float>(t, nullptr, 0); in AddInput()
100 id = AddTensor<float>(t, nullptr, 0, true); in AddVariableInput()
135 id = AddTensor<float>(t, nullptr, 0); in AddOutput()
H A Dtest_util.h214 id = AddTensor(t, data, size); in AddConstInput()
686 int AddTensor(TensorData t, const T* data, size_t size,
1092 return AddTensor<T>(t, {}, false);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op.cc72 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.indices(), &indices_node)); in AsGraphDefInternal()
74 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.values(), &value_node)); in AsGraphDefInternal()
H A Dpadded_batch_dataset_op.cc150 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
158 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
H A Dshuffle_dataset_op.cc612 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
671 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
922 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
H A Dcache_dataset_ops.cc714 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
1100 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsession_ops.cc53 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
/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/kernels/data/experimental/
H A Dset_stats_aggregator_dataset_op.cc160 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
H A Dscan_dataset_op.cc146 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
H A Dthreadpool_dataset_op.cc220 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
H A Dcsv_dataset_op.cc235 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
H A Dparse_example_dataset_op.cc302 TF_RETURN_IF_ERROR(b->AddTensor(dense_default, &node)); in AsGraphDefInternal()
H A Ddata_service_dataset_op.cc352 TF_RETURN_IF_ERROR(b->AddTensor(handle, &iteration_counter_handle)); in AsGraphDefInternal()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc924 return AddTensor(tensor_index, hybrid_op, &augmented_inputs_, tensor_flags); in AddTensorInput()
928 return AddTensor(tensor_index, /*hybrid_op=*/false, &augmented_outputs_, in AddTensorOutput()
1713 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder