Searched refs:c_dim (Results 1 – 2 of 2) sorted by relevance
431 int c_dim = GetTensorFeatureDimIndex(num_dims, attrs.data_format); in MakeXlaBackpropFilterConvOp() local432 int64 in_depth = input_shape.dimensions(c_dim), in MakeXlaBackpropFilterConvOp()443 dnums.set_input_batch_dimension(c_dim); in MakeXlaBackpropFilterConvOp()450 dnums.set_kernel_output_feature_dimension(c_dim); in MakeXlaBackpropFilterConvOp()
67 const Shape& shape, int64 n_dim, int64 c_dim, in GetShapeInfo() argument79 nchw_to_logical.push_back(c_dim); in GetShapeInfo()