| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops_test.cc | 129 const string& name, int32 id, int32 num_tensors) { in BuildNodeDef()
146 int32 num_tensors) { in BuildNodeDefWithReshape()
163 const string& name, int32 id, int32 num_tensors) { in MakeOp()
263 const string& name, int32 id, int32 num_tensors, in BuildNodeDef()
279 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 | 1249 int32 num_tensors = static_cast<int32>(tensor_lengths->NumElements()); in Compute() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_api_parameter_converter.cc | 268 Py_ssize_t num_tensors = PyList_GET_SIZE(tensor_list); in ConvertInputsWithTypeAttr() local
290 Py_ssize_t num_tensors = PyList_GET_SIZE(tensor_list); in ConvertInputsWithTypeAttr() local
331 Py_ssize_t num_tensors = PyList_GET_SIZE(params[input.tensor_list_params[0]]); in ConvertInputsWithTypeListAttr() local
415 int num_tensors = PyList_GET_SIZE(params[inputs.tensor_list_params[j]]); in InferLengthAttributes() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | portable_tensor.h | 38 int num_tensors = tensor_list.size; in VectorOfTensors() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.h | 173 size_t num_tensors() const { return tensors_.size(); } in num_tensors() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | memory_management.cc | 47 size_t num_tensors = obj_assignment.object_ids.size(); in ObjectsToOffsets() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/ |
| D | ethosu.cc | 70 int num_tensors = 0; in Eval() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | greedy_by_size_assignment.cc | 71 const size_t num_tensors = usage_records.size(); in GreedyBySizeAssignment() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | test_util.cc | 74 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.cc | 815 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors() function in tensorflow::InMemoryRunStepResponse
858 size_t OwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::OwnedProtoRunStepResponse
910 size_t NonOwnedProtoRunStepResponse::num_tensors() const { in num_tensors() function in tensorflow::NonOwnedProtoRunStepResponse
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | dumping_callback.py | 656 def _get_symbolic_tensor_ids(self, num_tensors): argument
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 63 size_t num_tensors() const override { return tensors_.size(); } in num_tensors() function in tflite::__anonb6ed36e90111::SimpleTestGraph
|
| D | arena_planner_test.cc | 143 size_t num_tensors() const override { return graph_->tensors()->size(); } in num_tensors() function in tflite::__anond3fd21620111::TestGraphInfo
|
| D | interpreter_test.cc | 1447 int num_tensors = 3; in SetUp() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 847 def num_tensors(self): member in TrackableWeightHandler
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 487 int64_t num_tensors = preceding_build_list.getNumOperands(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 1046 Py_ssize_t num_tensors = PySequence_Fast_GET_SIZE(tensors); in TFE_Py_TensorShapeSlice() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 390 int64 num_tensors = serialized_t->dim_size(0); in InitFromTensor() local
|
| D | cache_dataset_ops.cc | 129 static size_t StringPaddingSize(size_t num_tensors) { in StringPaddingSize()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_util.cc | 789 int num_tensors = metadata->tensor_metadata_size(); in SnappyUncompress() local
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | test_helpers.cc | 870 const Model* GetModelWithOfflinePlanning(int num_tensors, in GetModelWithOfflinePlanning()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 180 size_t num_tensors() const override { return subgraph_->tensors_size(); } in num_tensors() function in tflite::InterpreterInfo
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 138 const string& key_prefix, int64 src_incarnation, int64 num_tensors, in ReceiveTensorsAsync()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 927 const size_t num_tensors = op.N(); in Verify() local
|