| /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
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 …]
|
| 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()
|
| /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 | 717 return AddTensor(t, output); in AddDatasetOrTensor()
|
| D | dataset.h | 219 Status AddTensor(const Tensor& val, Node** output) { in AddTensor() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 89 id = AddTensor<float>(t, {}); in AddInput()
100 id = AddTensor<float>(t, {}, true); in AddVariableInput()
130 int id = AddTensor<float>(t, {}); in AddOutput()
|
| D | test_util.h | 206 id = AddTensor(t, data); in AddConstInput()
629 int AddTensor(TensorData t, std::initializer_list<T> data,
1016 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 | 150 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
158 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | shuffle_dataset_op.cc | 558 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
617 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
868 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()
1010 TF_RETURN_IF_ERROR(b->AddTensor(handle, &resource_handle_node)); in AsGraphDefInternal()
|
| /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 | 204 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
| D | csv_dataset_op.cc | 229 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | data_service_dataset_op.cc | 224 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 | 861 return AddTensor(tensor_index, hybrid_op, &augmented_inputs_, tensor_flags); in AddTensorInput()
865 return AddTensor(tensor_index, /*hybrid_op=*/false, &augmented_outputs_, in AddTensorOutput()
1423 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::delegate::nnapi::NNAPIOpBuilder
|