Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsplit_op.cc146 std::vector<int64> split_sizes; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1482 SmallVector<int64_t, 4> split_sizes; in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3019 SmallVector<int64_t, 4> split_sizes; in matchAndRewrite() local