Searched defs:axis_dim (Results 1 – 2 of 2) sorted by relevance
87 const int64 axis_dim = output_shape.dim_size(axis); in Compute() local
96 const Eigen::DenseIndex axis_dim = input_shape.dim_size(axis); in Compute() local