Searched refs:split_size_shape (Results 1 – 1 of 1) sorted by relevance
131 const TensorShape split_size_shape = ctx->InputShape(1); in Compile() local133 split_size_shape.dims() == 1 && in Compile()134 split_size_shape.num_elements() == num_split, in Compile()139 split_size_shape.dims(), "-D and ", in Compile()140 split_size_shape.num_elements(), " elements")); in Compile()