| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/ |
| H A D | IScheduler.h | 78 : _split_dimension(split_dimension), _strategy(strategy), _threshold(threshold) in _split_dimension() argument
87 Hints &set_split_dimension(unsigned int split_dimension) in set_split_dimension()
96 unsigned int split_dimension() const in split_dimension() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/helpers/ |
| H A D | WindowHelpers.cpp | 244 size_t split_dimension = Window::DimY; in calculate_squashed_or_max_window() local
293 size_t split_dimension = Window::DimY; in calculate_squashed_or_max_window() local
|
| /aosp_15_r20/build/soong/cmd/extract_apks/bundle_proto/ |
| H A D | config.proto | 181 repeated SplitDimension split_dimension = 1; field
187 repeated SplitDimension split_dimension = 1; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| H A D | xla_sharding.py | 170 def split(cls, tensor, split_dimension, num_devices, input_shape=None): argument
349 split_dimension, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | tpu_cross_replica_ops.cc | 44 int split_dimension; in __anon173a7a2a0102() local
|
| H A D | array_ops.cc | 575 DimensionHandle split_dimension; in __anon38bbb0e80a02() local
611 DimensionHandle split_dimension; in __anon38bbb0e80b02() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuAdd.cpp | 53 …const auto split_dimension = static_cast<kernels::CpuAddKernel *>(_kernel.get())->get_split_dimens… in run() local
|
| H A D | CpuSub.cpp | 53 …const auto split_dimension = static_cast<kernels::CpuSubKernel *>(_kernel.get())->get_split_dimens… in run() local
|
| H A D | CpuActivation.cpp | 53 …auto split_dimension = static_cast<kernels::CpuActivationKernel *>(_kernel.get())->get_split_dimen… in run() local
|
| H A D | CpuMul.cpp | 57 …auto split_dimension = static_cast<kernels::CpuMulKernel *>(_kernel.get())->get_split_dimension_hi… in run() local
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/ |
| H A D | IScheduler.cpp | 178 std::size_t IScheduler::adjust_num_of_windows(const Window &window, std::size_t split_dimension, st… in adjust_num_of_windows()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ops/ |
| H A D | tpu_ops.py | 42 split_dimension, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | xla_sharding_util.cc | 46 const int split_dimension, in CreateSplitOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expander_common.cc | 97 Status CreateSplitOp(const int num_split, const int split_dimension, in CreateSplitOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 3077 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64_t split_dimension, in AllToAll()
3091 XlaOp XlaBuilder::AllToAllArray(XlaOp operand, int64_t split_dimension, in AllToAllArray()
3183 XlaOp XlaBuilder::AllToAllTuple(XlaOp operand, int64_t split_dimension, in AllToAllTuple()
4690 XlaOp AllToAll(const XlaOp operand, int64_t split_dimension, in AllToAll()
4705 XlaOp AllToAllTuple(const XlaOp operand, int64_t split_dimension, in AllToAllTuple()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instructions.h | 730 std::optional<int64_t> split_dimension() const { return split_dimension_; } in split_dimension() function
|
| H A D | hlo_instruction.cc | 539 std::optional<int64_t> split_dimension; in CreateFromProto() local
1315 const std::optional<int64_t>& split_dimension) { in CreateAllToAll()
|
| H A D | shape_inference.cc | 2138 const Shape& shape, int64_t split_dimension, int64_t concat_dimension, in InferAllToAllShape()
|
| H A D | hlo_parser.cc | 1531 optional<int64_t> split_dimension; in CreateInstruction() local
|
| H A D | hlo_instructions.cc | 946 const std::optional<int64_t>& split_dimension) in HloAllToAllInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_util.cc | 1535 int64_t channel_id, std::optional<int64_t> split_dimension) { in GetPerGroupCollectiveOpsCreator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 838 int split_dimension = sharding_it->first; in CreateOrGetSplitNodesForInputSharding() local
|