Home
last modified time | relevance | path

Searched defs:split_sizes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsplit_op.cc147 std::vector<int64_t> split_sizes; in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dsplit_v_op.cc199 absl::Span<const Tlen> split_sizes) { in SplitHasAlignedOutputsInFirstDimension()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1597 SmallVector<int64_t, 4> split_sizes; in verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3511 SmallVector<int64_t, 4> split_sizes; in matchAndRewrite() local