Searched defs:output_batch_dimension (Results 1 – 2 of 2) sorted by relevance
219 int64 output_batch_dimension = dim_numbers.output_batch_dimension(); in HandleBatchGroupCount() local
504 int64 output_batch_dimension = 9; field