Searched defs:split_dim_size (Results 1 – 7 of 7) sorted by relevance
183 IndexType split_dim_size = in SetDims() local223 Eigen::DenseIndex split_dim_size, in operator ()()319 Eigen::DenseIndex split_dim_size; in Compute() local395 int32_t split_dim_size; in Compute() local441 Eigen::DenseIndex split_dim_size; in Compute() local
116 IndexType split_dim_size = in SetDims() local135 Eigen::DenseIndex split_dim_size, in operator ()()226 Eigen::DenseIndex split_dim_size; in Compute() local293 int32_t split_dim_size; in Compute() local
74 int32 prefix_dim_size, int32 split_dim_size, in SplitOpKernel()196 int32 prefix_dim_size, int32 split_dim_size, in Run()
219 const Index split_dim_size = dense_shape.dim_size(static_cast<int>(axis)); in operator ()() local
60 const int64_t split_dim_size = scatter_shape.dimensions(split_dim); in Run() local
515 const int split_dim_size = input_tensor.shape()[split_dim]; in Split() local
590 DimensionHandle split_dim_size; in __anon38bbb0e80a02() local675 auto split_dim_size = c->Value(c->Dim(input, split_dim)); in __anon38bbb0e80b02() local