Home
last modified time | relevance | path

Searched defs:tensor_list (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dportable_tensor.h37 const TfLiteIntArray& tensor_list) { in VectorOfTensors()
81 const TfLiteIntArray& tensor_list) in VectorOfQuantizedTensors()
/external/tensorflow/tensorflow/python/framework/
Dpython_api_parameter_converter.cc266 PyObject* tensor_list = params[index]; in ConvertInputsWithTypeAttr() local
288 PyObject* tensor_list = params[index]; in ConvertInputsWithTypeAttr() local
362 PyObject* tensor_list = params[index]; in ConvertInputsWithTypeListAttr() local
373 PyObject* tensor_list = params[index]; in ConvertInputsWithTypeListAttr() local
Dtype_spec.py242 def _from_tensor_list(self, tensor_list): argument
259 def _from_compatible_tensor_list(self, tensor_list): argument
338 def __check_tensor_list(self, tensor_list): argument
Dsparse_tensor.py341 def _from_components(self, tensor_list): argument
374 def _from_compatible_tensor_list(self, tensor_list): argument
Dindexed_slices.py260 def _from_components(self, tensor_list): argument
Dtensor_spec.py195 def _from_compatible_tensor_list(self, tensor_list): argument
Dcomposite_tensor_test.py56 def _from_components(self, tensor_list): argument
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py189 def _from_tensor_list_helper(decode_fn, element_spec, tensor_list): argument
224 def from_compatible_tensor_list(element_spec, tensor_list): argument
247 def from_tensor_list(element_spec, tensor_list): argument
/external/tensorflow/tensorflow/python/training/
Dinput.py328 def slice_input_producer(tensor_list, num_epochs=None, shuffle=True, seed=None, argument
466 def _as_original_type(original_tensors, tensor_list): argument
478 def _store_sparse_tensors(tensor_list, enqueue_many, keep_input, argument
630 def _validate(tensor_list): argument
712 def _select_which_to_enqueue(tensor_list, keep_input): argument
738 def _enqueue(queue, tensor_list, threads, enqueue_many, keep_input): argument
/external/tensorflow/tensorflow/python/autograph/lang/
Dspecial_functions.py57 def tensor_list(elements, function
/external/tensorflow/tensorflow/python/keras/tests/
Dgraph_util_test.py37 def _get_tensors(self, sess, tensor_list): argument
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h69 const TensorList* tensor_list = nullptr; in Compute() local
297 const TensorList* tensor_list = nullptr; in Compute() local
557 const TensorList* tensor_list = nullptr; in Compute() local
Dlist_kernels.cc66 const TensorList& tensor_list, int index, in GetElementShapeFromInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression.cc59 XlaExpression XlaExpression::TensorList(xla::XlaOp tensor_list) { in TensorList()
Dgraph_compiler.cc101 const xla::XlaOp& tensor_list = expressions[i]->handle(); in PrepareArguments() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc202 if (auto tensor_list = arg_type.dyn_cast<TFRTensorListType>()) { in Verify() local
279 if (auto tensor_list = result_type.dyn_cast<TFRTensorListType>()) { in Verify() local
/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py865 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
Dtensor_array_ops.py1383 def _from_components(self, tensor_list): argument
Dresource_variable_ops.py2261 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc389 SmallVector<Value, 4> tensor_list; in CreateAndReplaceOp() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2374 def _from_components(self, tensor_list): argument
2433 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor.py1114 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/python/client/
Dsession.py583 def _name_list(tensor_list): argument
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py401 def _from_components(self, tensor_list): argument