/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
H A D | verifier_test.cc | 83 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 D | verifier_internal_test.cc | 61 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 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 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 D | subgraph.h | 45 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 D | operation_selector.cc | 106 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 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 82 TF_RETURN_IF_ERROR(session_state->AddTensor(key, it->second.tensor)); in SaveTensors()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | session_state.h | 37 Status AddTensor(const std::string& handle, const Tensor& tensor); 71 Status AddTensor(const std::string& name, const TensorAndKey& tk);
|
H A D | dataset.h | 224 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
|
H A D | dataset.cc | 803 return AddTensor(t, output); in AddDatasetOrTensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | test_util.cc | 89 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 D | test_util.h | 214 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 D | sparse_tensor_slice_dataset_op.cc | 72 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 D | padded_batch_dataset_op.cc | 150 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal() 158 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
H A D | shuffle_dataset_op.cc | 612 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 D | cache_dataset_ops.cc | 714 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 D | session_ops.cc | 53 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
|
/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/kernels/data/experimental/ |
H A D | set_stats_aggregator_dataset_op.cc | 160 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
H A D | scan_dataset_op.cc | 146 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
H A D | threadpool_dataset_op.cc | 220 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
H A D | csv_dataset_op.cc | 235 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
H A D | parse_example_dataset_op.cc | 302 TF_RETURN_IF_ERROR(b->AddTensor(dense_default, &node)); in AsGraphDefInternal()
|
H A D | data_service_dataset_op.cc | 352 TF_RETURN_IF_ERROR(b->AddTensor(handle, &iteration_counter_handle)); in AsGraphDefInternal()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | nnapi_delegate.cc | 924 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
|