Searched defs:n_dim (Results 1 – 6 of 6) sorted by relevance
67 const Shape& shape, int64 n_dim, int64 c_dim, in GetShapeInfo()
430 int n_dim = GetTensorBatchDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local
1057 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local1184 TensorShapeProto_Dim n_dim; in GenerateBatchMatmulContextFromEinsum() local1840 int64 n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul() local
1362 DimensionHandle n_dim; in __anon4650176f2002() local
90 DimensionHandle n_dim = c->Dim(paddings, 0); in PadShapeFn() local
6618 unsigned int basic_set::n_dim() const in n_dim() function15083 unsigned int set::n_dim() const in n_dim() function