Searched defs:n_dims (Results 1 – 6 of 6) sorted by relevance
51 const int64_t n_dims = shape.rank(); in SliceInMinorDims() local76 const int64_t n_dims = shape.rank(); in UpdateSlice() local96 const int64_t n_dims = shape.rank(); in UpdateSliceInMinorDims() local120 const int64_t n_dims = shape.rank(); in PrependZerosInMajorDims() local136 const int64_t n_dims = shape.rank(); in DynamicSliceInMinorDims() local
62 auto n_dims = static_cast<int32_t>(shape.rank()); in GetDiagonalMask() local80 auto n_dims = static_cast<int32_t>(shape.rank()); in GetMatrixDiagonal() local114 auto n_dims = static_cast<int32_t>(shape.rank()); in GetMatrixDiagonalViaGather() local178 auto n_dims = static_cast<int32_t>(shape.rank()); in SetMatrixDiagonal() local216 const int64_t n_dims = shape.rank(); in TriangleMask() local738 const int64_t n_dims = shape.rank(); in TransposeInMinorDims() local
4 const int n_dims = 5; variable
44 const int32_t n_dims = static_cast<int32_t>(src.shape().num_dimensions()); in softmax_layer_generic() local
492 SaveArrayAsNumpy(const std::string &filename, bool fortran_order, unsigned int n_dims, const unsign… in SaveArrayAsNumpy()513 SaveArrayAsNumpy(const std::string &filename, bool fortran_order, unsigned int n_dims, const unsign… in SaveArrayAsNumpy()
1518 int n_dims; in CheckAndFetchProjectionWeights() local1609 int n_dims; in Create() local