Searched defs:outer_dims (Results 1 – 9 of 9) sorted by relevance
53 const unsigned int outer_dims = dst.shape().total_size() / (W * H * C); in reorg_layer() local
26 std::vector<int64_t> outer_dims; in Run() local
2018 llvm::SmallVector<int64_t> outer_dims; in HandleSlice() local2633 std::vector<int64_t> outer_dims(std::next(concat_dim_layout_itr), in EmitFastConcatenate() local
58 const int outer_dims = indices.DimensionsCount() - 1; in CheckShapes() local
96 const int64_t outer_dims = indices.shape().dims() - 1; in Compute() local169 const int64_t outer_dims = indices.shape().dims() - 1; in Compute() local
669 const int outer_dims = indices_shape.DimensionsCount() - 1; in ScatterNd() local
836 const auto outer_dims = output_type.getShape().take_front(axis); in ConstFoldConcatenateOpDense() local1427 auto outer_dims = indices_type.getRank() - 1; in verify() local
2427 const int64_t outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
2733 const int64_t outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local