Searched defs:axis_dim (Results 1 – 3 of 3) sorted by relevance
62 StatusOr<int64_t> axis_dim = GetAxisDimension(op); in ExpandOp() local
87 const int64_t axis_dim = output_shape.dim_size(axis); in Compute() local
95 const Eigen::DenseIndex axis_dim = input_shape.dim_size(axis); in Compute() local