Searched defs:outer_dims (Results 1 – 9 of 9) sorted by relevance
25 std::vector<int64> outer_dims; in Run() local
1907 std::vector<int64> outer_dims; in HandleSlice() local2472 std::vector<int64> outer_dims(std::next(concat_dim_layout_itr), in EmitFastConcatenate() local
58 const int outer_dims = indices.DimensionsCount() - 1; in CheckShapes() local
112 llvm::SmallVector<int64_t, 5> outer_dims; in ProcessDotArg() local
95 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local168 const int64 outer_dims = indices.shape().dims() - 1; in Compute() local
704 const auto outer_dims = output_type.getShape().take_front(axis); in ConstFoldConcatenateOpDense() local1045 auto outer_dims = indices_type.getRank() - 1; in Verify() local
2335 const int64 outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
1147 const int outer_dims = indices_shape.DimensionsCount() - 1; in ScatterNd() local
2443 const int64 outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local