Home
last modified time | relevance | path

Searched defs:spatial_dimensions (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.h28 absl::Span<const int64> spatial_dimensions) in TensorFormat()
Dpooling_test.cc26 absl::InlinedVector<int64, 4> spatial_dimensions; in MakeNCHWFormat() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc140 absl::InlinedVector<int64, 4> spatial_dimensions(num_spatial_dims); in XlaTensorFormat() local