Home
last modified time | relevance | path

Searched refs:feature_dimension (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.h27 TensorFormat(int batch_dimension, int feature_dimension, in TensorFormat() argument
30 feature_dimension_(feature_dimension), in TensorFormat()
36 int feature_dimension() const { return feature_dimension_; } in feature_dimension() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc139 int feature_dimension = GetTensorFeatureDimIndex(num_dims, data_format); in XlaTensorFormat() local
146 /*feature_dimension=*/feature_dimension, in XlaTensorFormat()