Home
last modified time | relevance | path

Searched defs:size_split (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_tensor_ops.py774 def size_split(N, i): function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tf.cc1572 SmallVector<int32_t> size_split; in matchAndRewrite() local
H A Dlegalize_common.cc2098 Value input_value, SmallVectorImpl<int32_t>& size_split, int32_t axis) { in convertSplitVOp()
H A Dlegalize_tfl.cc2191 SmallVector<int32_t> size_split; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc2445 int64_t size_split = size_split_attr.getValue().getSExtValue(); in verify() local