Searched refs:use_batch_group_count (Results 1 – 1 of 1) sorted by relevance
441 bool use_batch_group_count = in MakeXlaBackpropFilterConvOp() local460 if (use_batch_group_count) { in MakeXlaBackpropFilterConvOp()536 /*batch_group_count=*/use_batch_group_count ? dims.in_depth : 1); in MakeXlaBackpropFilterConvOp()538 if (!use_batch_group_count && attrs.depthwise) { in MakeXlaBackpropFilterConvOp()