Home
last modified time | relevance | path

Searched defs:batch_dimension (Results 1 – 3 of 3) 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.cc140 int batch_dimension = GetTensorBatchDimIndex(num_dims, data_format); in XlaTensorFormat() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc570 HloInstruction* activations, int64 batch_dimension, int64 old_batch_size, in IncreaseSpatialSizeOnSpaceToBatchedShape()