Searched refs:is_output_batch_dim (Results 1 – 1 of 1) sorted by relevance
788 bool is_output_batch_dim = in IterationSpaceForOutputBatchIndices() local790 index_count.push_back(is_output_batch_dim ? output_shape.dimensions(i) : 1); in IterationSpaceForOutputBatchIndices()