Searched defs:split_dim_size (Results 1 – 6 of 6) sorted by relevance
173 IndexType split_dim_size = in SetDims() local193 Eigen::DenseIndex split_dim_size, in operator ()()284 Eigen::DenseIndex split_dim_size; in Compute() local360 int32 split_dim_size; in Compute() local406 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 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()
514 const int split_dim_size = input_tensor.shape()[split_dim]; in Split() local
589 DimensionHandle split_dim_size; in __anondb9326b20a02() local673 auto split_dim_size = c->Value(c->Dim(input, split_dim)); in __anondb9326b20b02() local
682 auto split_dim_size = type.getDimSize(op.split_dimension()); in Verify() local