| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 83 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
177 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
178 builder.AddTensor({}, TensorType_UINT8, {}, "empty_vector"); in TEST()
179 builder.AddTensor( in TEST()
183 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
191 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
192 builder.AddTensor( in TEST()
196 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
225 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
226 builder.AddTensor( in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | session_state.h | 37 Status AddTensor(const std::string& handle, const Tensor& tensor);
71 Status AddTensor(const std::string& name, const TensorAndKey& tk);
|
| D | dataset.cc | 501 return AddTensor(t, output); in AddDatasetOrTensor()
|
| D | dataset.h | 192 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | test_helpers.cc | 92 Tensor AddTensor(TensorType type, std::initializer_list<int32_t> shape) { in AddTensor() function in tflite::testing::__anon4d49c9e80111::ModelBuilder
264 const int input_tensor = model_builder.AddTensor(TensorType_UINT8, {3}); in BuildSimpleStatefulModel()
265 const int median_tensor = model_builder.AddTensor(TensorType_UINT8, {3}); in BuildSimpleStatefulModel()
267 model_builder.AddTensor(TensorType_INT32, {1}); in BuildSimpleStatefulModel()
305 const int t0 = model_builder.AddTensor(TensorType_FLOAT32, {2, 2, 3}); in BuildSimpleModelWithBranch()
306 const int t1 = model_builder.AddTensor(TensorType_FLOAT32, {2, 2, 3}); in BuildSimpleModelWithBranch()
307 const int t2 = model_builder.AddTensor(TensorType_FLOAT32, {2, 2, 3}); in BuildSimpleModelWithBranch()
308 const int t3 = model_builder.AddTensor(TensorType_FLOAT32, {2, 2, 3}); in BuildSimpleModelWithBranch()
329 model_builder.AddTensor(TensorType_FLOAT32, {2, 2, 3}); in BuildModelWithOfflinePlanning()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 88 id = AddTensor<float>(t, {}); in AddInput()
99 id = AddTensor<float>(t, {}, true); in AddVariableInput()
130 int id = AddTensor<float>(t, {}); in AddOutput()
|
| D | test_util.h | 205 id = AddTensor(t, data); in AddConstInput()
630 int AddTensor(TensorData t, std::initializer_list<T> data,
1017 return AddTensor<T>(t, {}, false);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 70 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.indices(), &indices_node)); in AsGraphDefInternal()
72 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.values(), &value_node)); in AsGraphDefInternal()
|
| D | padded_batch_dataset_op.cc | 148 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
156 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | shuffle_dataset_op.cc | 534 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
593 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
844 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
|
| D | cache_dataset_ops.cc | 651 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
1006 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
|
| D | captured_function.cc | 578 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AddToGraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 53 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | set_stats_aggregator_dataset_op.cc | 158 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
| D | scan_dataset_op.cc | 138 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | threadpool_dataset_op.cc | 193 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
| D | csv_dataset_op.cc | 223 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | data_service_dataset_op.cc | 201 TF_RETURN_IF_ERROR(b->AddTensor(handle, &iteration_counter_handle)); in AsGraphDefInternal()
|
| D | parse_example_dataset_op.cc | 300 TF_RETURN_IF_ERROR(b->AddTensor(dense_default, &node)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 786 return AddTensor(tensor_index, hybrid_op, &augmented_inputs_, tensor_flags); in AddTensorInput()
790 return AddTensor(tensor_index, /*hybrid_op=*/false, &augmented_outputs_, in AddTensorOutput()
1257 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder
|