Searched defs:n_dim (Results 1 – 5 of 5) sorted by relevance
430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
68 const Shape& shape, int64_t n_dim, int64_t c_dim, in GetShapeInfo()
1068 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local1195 TensorShapeProto_Dim n_dim; in GenerateBatchMatmulContextFromEinsum() local1853 int64_t n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
1362 DimensionHandle n_dim; in __anon97b5e01b2002() local
90 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local