Searched defs:split_dim_size (Results 1 – 5 of 5) sorted by relevance
119 IndexType split_dim_size = in SetDims() local138 Eigen::DenseIndex split_dim_size, in operator ()()229 Eigen::DenseIndex split_dim_size; in Compute() local296 int32 split_dim_size; in Compute() local357 Eigen::DenseIndex split_dim_size; in Compute() local
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
79 int32 split_dim_size, int32 suffix_dim_size, in SplitOpKernel()199 int32 prefix_dim_size, int32 split_dim_size, in Run()
591 const int split_dim_size = input_tensor.shape()[split_dim]; in Split() local
583 DimensionHandle split_dim_size; in __anondb9326b20a02() local667 auto split_dim_size = c->Value(c->Dim(input, split_dim)); in __anondb9326b20b02() local