Searched refs:feature_dimension (Results 1 – 2 of 2) sorted by relevance
27 TensorFormat(int batch_dimension, int feature_dimension, in TensorFormat() argument30 feature_dimension_(feature_dimension), in TensorFormat()36 int feature_dimension() const { return feature_dimension_; } in feature_dimension() function
139 int feature_dimension = GetTensorFeatureDimIndex(num_dims, data_format); in XlaTensorFormat() local146 /*feature_dimension=*/feature_dimension, in XlaTensorFormat()