Searched defs:feature_dimension (Results 1 – 2 of 2) sorted by relevance
27 TensorFormat(int batch_dimension, int 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() local