Searched defs:n_dim (Results 1 – 10 of 10) sorted by relevance
15 int n_dim = static_cast<int>(sizes.size()); in possible_cross_dimension_overlap() local154 int n_dim = static_cast<int>(sizes.size()); in computeStrideProps() local
43 int n_dim = static_cast<int>(sizes.size()); in is_contiguous_strides() local
11 size_t n_dim = t.dim(); in CheckStrideIndices() local
117 const size_t n_dim = self.dim() - 1; in resize_out_tensor() local
1069 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local1196 TensorShapeProto_Dim n_dim; in GenerateBatchMatmulContextFromEinsum() local1874 int64_t n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
69 const Shape& shape, int64_t n_dim, int64_t c_dim, in GetShapeInfo()
203 auto n_dim = negative.dim(); in triplet_margin_loss() local
1491 DimensionHandle n_dim; in __anon84cbdd652102() local
91 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local