Home
last modified time | relevance | path

Searched defs:tensor_list (Results 1 – 25 of 25) 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
Dsparse_tensor.py338 def _from_components(self, tensor_list): argument
372 def _from_compatible_tensor_list(self, tensor_list): argument
Dtype_spec.py513 def __check_tensor_list(self, tensor_list): argument
861 def batchable_from_tensor_list(spec, tensor_list): argument
Dindexed_slices.py270 def _from_components(self, tensor_list): argument
Dcomposite_tensor_test.py52 def _from_components(self, tensor_list): argument
Dtensor_spec.py206 def _from_compatible_tensor_list(self, tensor_list): argument
Dextension_type.py681 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py193 def _from_tensor_list_helper(decode_fn, element_spec, tensor_list): argument
228 def from_compatible_tensor_list(element_spec, tensor_list): argument
251 def from_tensor_list(element_spec, tensor_list): argument
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
Dtensor_list_dynamic_shape.py51 def loop_body(i, tensor_list): argument
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dtensor_list_dynamic_shape.py47 def loop_body(i, tensor_list): argument
/external/tensorflow/tensorflow/python/training/
Dinput.py323 def slice_input_producer(tensor_list, num_epochs=None, shuffle=True, seed=None, argument
461 def _as_original_type(original_tensors, tensor_list): argument
473 def _store_sparse_tensors(tensor_list, enqueue_many, keep_input, argument
625 def _validate(tensor_list): argument
709 def _select_which_to_enqueue(tensor_list, keep_input): argument
735 def _enqueue(queue, tensor_list, threads, enqueue_many, keep_input): argument
/external/tensorflow/tensorflow/python/autograph/lang/
Dspecial_functions.py53 def tensor_list(elements, function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression.cc60 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/core/kernels/
Dlist_kernels.h174 const TensorList* tensor_list = nullptr; in Compute() local
403 const TensorList* tensor_list = nullptr; in Compute() local
669 const TensorList* tensor_list = nullptr; in Compute() local
Dlist_kernels.cc71 const TensorList& tensor_list, int index, in GetElementShapeFromInput()
/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py876 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
Dtensor_array_ops.py1441 def _from_components(self, tensor_list): argument
Dresource_variable_ops.py2398 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc227 if (auto tensor_list = arg_type.dyn_cast<TFRTensorListType>()) { in verify() local
304 if (auto tensor_list = result_type.dyn_cast<TFRTensorListType>()) { in verify() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc405 SmallVector<Value, 4> tensor_list; in CreateAndReplaceOp() local
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py2503 def _from_components(self, tensor_list): argument
2584 def _from_compatible_tensor_list(self, tensor_list): argument
/external/tensorflow/tensorflow/python/client/
Dsession.py580 def _name_list(tensor_list): argument
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py409 def _from_components(self, tensor_list): argument