HomeSort by relevance Sort by last modified time
    Searched refs:split_dim_tensor (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
split_op_test.cc 36 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); local
39 .Input(test::graph::Constant(g, split_dim_tensor))
split_v_op_test.cc 63 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); local
69 .Input(test::graph::Constant(g, split_dim_tensor))
split_op.cc 53 const Tensor& split_dim_tensor = context->input(0); local
55 context, split_dim_tensor.shape().dims() == 0,
57 split_dim_tensor.shape().dims()));
58 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0);
split_v_op.cc 59 const Tensor& split_dim_tensor = context->input(2); local
61 OP_REQUIRES(context, split_dim_tensor.NumElements() == 1,
62 errors::InvalidArgument("split_dim_tensor must have "
65 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0);

Completed in 1408 milliseconds