Searched refs:kConvBatchDimension (Results 1 – 2 of 2) sorted by relevance
2648 dimension_numbers.set_input_batch_dimension(kConvBatchDimension); in CreateDefaultConvDimensionNumbers()2650 dimension_numbers.set_output_batch_dimension(kConvBatchDimension); in CreateDefaultConvDimensionNumbers()
176 static constexpr int64 kConvBatchDimension = 0; variable