| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A 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,
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 3961 def index_copy(x: TensorLike, dim: int, index: TensorLike, tensor: TensorLike):
3967 def index_copy_(x: TensorLike, dim: int, index: TensorLike, tensor: TensorLike):
3983 x: TensorLike, dim: int, index: TensorLike, value: Union[NumberType, TensorLike] argument
3990 x: TensorLike, dim: int, index: TensorLike, value: Union[NumberType, TensorLike] argument
3996 x: TensorLike,
3998 index: TensorLike,
3999 value: Union[NumberType, TensorLike], argument
4043 x: TensorLike,
4045 index: TensorLike,
4046 tensor: TensorLike,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2570 x: TensorLike,
2572 index: TensorLike,
2573 tensor: TensorLike,
2583 x: TensorLike,
2585 index: TensorLike,
2586 tensor: TensorLike,
2594 x: TensorLike,
2596 index: TensorLike,
2597 tensor: TensorLike,
2661 def index_copy_(x: TensorLike, dim: int, index: TensorLike, tensor: TensorLike):
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| H A D | representative_dataset_test.py | 44 tensorlike_value_1: core.TensorLike,
45 tensorlike_value_2: core.TensorLike) -> None:
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A 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,
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 2947 input: TensorLike,
2966 input: TensorLike,
2990 input: TensorLike,
3006 input: TensorLike,
3030 input: TensorLike,
3046 input: TensorLike,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/types/ |
| H A D | core.py | 257 TensorLike = Union[Tensor, TensorProtocol, int, float, bool, str, bytes, variable
|
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/ |
| H A D | type_utils.py | 33 class TensorLike(Protocol): class
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | __init__.py | 61 TensorLike = (_Tensor, torch.Tensor) variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_array_ops.py | 1168 value: core_types.TensorLike,
|
| H A D | dynamic_ragged_shape.py | 198 inner_shape: core.TensorLike,
|
| /aosp_15_r20/external/pytorch/torch/_prims_common/ |
| H A D | __init__.py | 92 TensorLike = torch.Tensor variable
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_overrides.py | 359 class TensorLike: class
|