Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/signal/
H A Dfft_ops_test.py164 def test_empty(self, rank, extra_dims, np_type): argument
173 def test_basic(self, rank, extra_dims, np_type): argument
182 def test_large_batch(self, rank, extra_dims, np_type): argument
201 def test_placeholder(self, rank, extra_dims, np_type): argument
213 def test_random(self, rank, extra_dims, np_type): argument
260 def test_grad_simple(self, rank, extra_dims, np_type): argument
273 def test_grad_random(self, rank, extra_dims, np_type): argument
352 def test_empty(self, rank, extra_dims, np_rtype): argument
362 def test_basic(self, rank, extra_dims, size, np_rtype): argument
380 def test_large_batch(self, rank, extra_dims, size, np_rtype): argument
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesViews.cpp469 auto extra_dims = size.size() - (self_dim - 1); in expand_batch_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchingRegistrations.cpp181 auto extra_dims = size_physical.size() - self_physical_dim; in expand_batching_rule() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DSpectralOps.cpp899 const auto extra_dims = std::max(size_t{3}, input_dim) - input_dim; in stft() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2404 std::vector<int> extra_dims; in DynamicExpandDims() local