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