Searched defs:outer_dims (Results 1 – 10 of 10) sorted by relevance
53 const unsigned int outer_dims = dst.shape().total_size() / (W * H * C); in reorg_layer() local
25 std::vector<int64> outer_dims; in Run() local
1912 std::vector<int64> outer_dims; in HandleSlice() local2495 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
99 llvm::SmallVector<int64_t, 5> outer_dims; in ProcessDotArg() local
95 const int64_t outer_dims = indices.shape().dims() - 1; in Compute() local168 const int64_t outer_dims = indices.shape().dims() - 1; in Compute() local
764 const auto outer_dims = output_type.getShape().take_front(axis); in ConstFoldConcatenateOpDense() local1284 auto outer_dims = indices_type.getRank() - 1; in Verify() local
2375 const int64_t outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
1028 const int outer_dims = indices_shape.DimensionsCount() - 1; in ScatterNd() local
2505 const int64_t outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local