| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops_test.cc | 128 const string& name, int32 id, int32 num_tensors) { in BuildNodeDef()
145 int32 num_tensors) { in BuildNodeDefWithReshape()
162 const string& name, int32 id, int32 num_tensors) { in MakeOp()
262 const string& name, int32 id, int32 num_tensors, in BuildNodeDef()
278 int32 id, int32 num_tensors, in MakeOp()
|
| D | save_restore_v2_ops.cc | 45 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in ValidateInputs() local
103 const int num_tensors = static_cast<int>(tensor_names.NumElements()); in Compute() local
|
| D | tensor_array_ops.cc | 1257 int32 num_tensors = static_cast<int32>(tensor_lengths->NumElements()); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor.h | 52 int num_tensors = tensor_list.size; in VectorOfTensors() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 70 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors()
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 42 size_t num_tensors() const override { return tensors_.size(); } in num_tensors() function in tflite::__anon531b66cb0111::SimpleTestGraph
|
| D | arena_planner_test.cc | 132 size_t num_tensors() const override { return graph_->tensors()->size(); } in num_tensors() function in tflite::__anon43acde040111::TestGraphInfo
|
| D | interpreter_test.cc | 1545 int num_tensors = 3; in SetUp() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.cc | 796 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors() function in tensorflow::InMemoryRunStepResponse
839 size_t OwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::OwnedProtoRunStepResponse
891 size_t NonOwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 997 Py_ssize_t num_tensors = PySequence_Fast_GET_SIZE(tensors); in TFE_Py_TensorShapeSlice() local
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 114 size_t num_tensors() const override { return subgraph_->tensors().size(); } in num_tensors() function in tflite::InterpreterInfo
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | cache_dataset_ops.cc | 103 static size_t StringPaddingSize(size_t num_tensors) { in StringPaddingSize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 131 const string& key_prefix, int64 src_incarnation, int64 num_tensors, in ReceiveTensorsAsync()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 208 def _create_tensor_values_cache(graph, num_tensors): argument
|