Searched defs:single_dim (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| H A D | xla_sharding.py | 450 single_dim=-1, argument
479 single_dim=-1, argument
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | reference.py | 41 def _dims(d, N, keepdim, single_dim): argument
330 orig, dim_offset=0, keepdim_offset=1, dim_name="dim", single_dim=False, reduce=True argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 1049 int32 single_dim; in __anon6b9e6b800f02() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 5017 OpSharding GetManualSharding(const OpSharding& original, int64_t single_dim) { in GetManualSharding()
5061 XlaBuilder* builder, XlaOp input, int single_dim, in ConvertSpmdFullToShardShape()
5106 XlaBuilder* builder, XlaOp input, const Shape& output_shape, int single_dim, in ConvertSpmdShardToFullShape()
|
| /aosp_15_r20/external/pytorch/functorch/csrc/dim/ |
| H A D | dim.cpp | 2873 bool single_dim = false; member
|