Searched defs:TensorLike (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 275 def _control_input(self, control_input: Union[core.TensorLike,
334 input_tensor: core.TensorLike,
335 control_input: Optional[Union[core.TensorLike, ops.Operation]] = None,
363 def _all_gather(self, input_tensor: core.TensorLike,
390 input_tensor_packs: List[List[core.TensorLike]],
436 input_tensor: core.TensorLike,
437 axis: core.TensorLike,
533 [core.TensorLike, Optional[collective_util.Options]], core.Tensor]
554 input_tensor: core.TensorLike,
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| D | representative_dataset_test.py | 44 tensorlike_value_1: core.TensorLike,
45 tensorlike_value_2: core.TensorLike) -> None:
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 183 body: Callable[..., Union[core_types.TensorLike, Iterable]], # pylint:disable=g-bare-generic
184 inputs: Optional[List[core_types.TensorLike]] = None,
|
| /external/tensorflow/tensorflow/python/types/ |
| D | core.py | 257 TensorLike = Union[Tensor, TensorProtocol, int, float, bool, str, bytes, variable
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_array_ops.py | 1168 value: core_types.TensorLike,
|
| D | dynamic_ragged_shape.py | 198 inner_shape: core.TensorLike,
|