Searched defs:input_batch_dimension (Results 1 – 2 of 2) sorted by relevance
218 int64 input_batch_dimension = dim_numbers.input_batch_dimension(); in HandleBatchGroupCount() local
481 int64 input_batch_dimension = 7; field