Home
last modified time | relevance | path

Searched defs:batch_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()
34 int batch_dimension() const { return batch_dimension_; } in batch_dimension() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc138 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat() local