Searched defs:batch_dimension (Results 1 – 3 of 3) sorted by relevance
27 TensorFormat(int batch_dimension, int feature_dimension, in TensorFormat()34 int batch_dimension() const { return batch_dimension_; } in batch_dimension() function
148 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat() local
744 HloInstruction* activations, int64_t batch_dimension, in SplitAndTransposeMergedBatch()796 HloInstruction* activations, int64_t batch_dimension, in ChangeSpatialSizeOnSpaceToBatchedShape()