| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 74 void AddTensor(const std::vector<int>& shape, tflite::TensorType type, in AddTensor() function in tflite::TfLiteFlatbufferModelBuilder
154 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
155 builder.AddTensor({}, TensorType_UINT8, {}, "empty_vector"); in TEST()
156 builder.AddTensor( in TEST()
160 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
168 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4, 5, 6}, "input"); in TEST()
169 builder.AddTensor( in TEST()
173 builder.AddTensor({2, 3}, TensorType_INT32, {}, "output"); in TEST()
219 builder.AddTensor({2, 3}, TensorType_UINT8, {1, 2, 3, 4}, "input"); in TEST()
229 builder.AddTensor({2, 1}, TensorType_UINT8, {1, 2, 3, 4}, "input"); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | session_state.cc | 34 Status SessionState::AddTensor(const string& handle, const Tensor& tensor) { in AddTensor() function in tensorflow::SessionState
57 Status TensorStore::AddTensor(const string& name, const TensorAndKey& tk) { in AddTensor() function in tensorflow::TensorStore
78 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 string& handle, const Tensor& tensor);
71 Status AddTensor(const string& name, const TensorAndKey& tk);
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 54 id = AddTensor<float>(t, {}, is_variable); in AddInput()
67 int id = AddTensor<float>(t, {}); in AddOutput()
|
| D | test_util.h | 167 int id = AddTensor(TensorData{type, shape}, data); in AddConstInput()
381 int AddTensor(TensorData t, std::initializer_list<T> data,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | sparse_tensor_slice_dataset_op.cc | 64 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.indices(), &indices_node)); in AsGraphDefInternal()
66 TF_RETURN_IF_ERROR(b->AddTensor(sparse_tensor_.values(), &value_node)); in AsGraphDefInternal()
|
| D | padded_batch_dataset_op.cc | 181 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
189 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | tensor_dataset_op.cc | 85 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | tensor_slice_dataset_op.cc | 110 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | flat_map_dataset_op.cc | 103 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | filter_dataset_op.cc | 145 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | parallel_map_dataset_op.cc | 156 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | map_dataset_op.cc | 154 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | interleave_dataset_op.cc | 129 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | scan_dataset_op.cc | 113 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
127 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | set_stats_aggregator_dataset_op.cc | 142 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
| D | take_while_dataset_op.cc | 148 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | threadpool_dataset_op.cc | 182 TF_RETURN_IF_ERROR(b->AddTensor(resource_handle_, &resource_handle_node)); in AsGraphDefInternal()
|
| D | parse_example_dataset_op.cc | 225 TF_RETURN_IF_ERROR(b->AddTensor(dense_default, &node)); in AsGraphDefInternal()
|
| D | group_by_reducer_dataset_op.cc | 422 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in OtherArgumentsNodeAndType()
|
| D | choose_fastest_branch_dataset_op.cc | 271 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| D | group_by_window_dataset_op.cc | 507 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in OtherArgumentsNodeAndType()
|
| D | csv_dataset_op.cc | 189 TF_RETURN_IF_ERROR(b->AddTensor(t, &node)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 47 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 275 return AddTensor(tensor_index, hybrid_op, &augmented_inputs_); in AddTensorInput()
279 return AddTensor(tensor_index, /*hybrid_op=*/false, &augmented_outputs_); in AddTensorOutput()
404 TfLiteStatus AddTensor(int tensor_index, bool hybrid_op, in AddTensor() function in tflite::__anon03cbe2c10111::NNAPIOpBuilder
|