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.cc146 std::vector<int64> split_sizes; in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dsplit_v_op.cc192 absl::Span<const Tlen> split_sizes) { in SplitHasAlignedOutputsInFirstDimension()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1537 SmallVector<int64_t, 4> split_sizes; in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3703 SmallVector<int64_t, 4> split_sizes; in matchAndRewrite() local