Searched refs:split_dim_tensor (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsplit_op_test.cc36 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); local
39 .Input(test::graph::Constant(g, split_dim_tensor))
H A Dsplit_v_op_test.cc63 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); local
69 .Input(test::graph::Constant(g, split_dim_tensor))

Completed in 429 milliseconds