/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
|