Searched refs:split_dim_tensor (Results 1 – 4 of 4) sorted by relevance
36 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph() local39 .Input(test::graph::Constant(g, split_dim_tensor)) in MakeGraph()
63 Tensor split_dim_tensor = test::AsScalar<int32>(split_dim); in MakeGraph() local69 .Input(test::graph::Constant(g, split_dim_tensor)) in MakeGraph()
53 const Tensor& split_dim_tensor = context->input(0); in ComputeEasyCases() local55 context, split_dim_tensor.shape().dims() == 0, in ComputeEasyCases()57 split_dim_tensor.shape().dims())); in ComputeEasyCases()58 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases()
59 const Tensor& split_dim_tensor = context->input(2); in ComputeEasyCases() local61 OP_REQUIRES(context, split_dim_tensor.NumElements() == 1, in ComputeEasyCases()65 const int32 split_dim_orig = split_dim_tensor.flat<int32>()(0); in ComputeEasyCases()