Searched defs:batch_dimension (Results 1 – 2 of 2) 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
138 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat() local