Home
last modified time | relevance | path

Searched defs:spatial_dims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py328 spatial_dims=None, argument
505 spatial_dims=None, argument
657 def _with_space_to_batch_adjust(orig, fill_value, spatial_dims): argument
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc354 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()
378 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc676 const std::vector<int64>& spatial_dims) { in ImageDims()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc572 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc614 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor() local