Searched refs:extra_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 177 def test_empty(self, rank, extra_dims, np_type): argument
178 dims = rank + extra_dims
186 def test_basic(self, rank, extra_dims, np_type): argument
190 dims = rank + extra_dims
198 def test_large_batch(self, rank, extra_dims, np_type): argument
199 dims = rank + extra_dims
217 def test_placeholder(self, rank, extra_dims, np_type): argument
221 dims = rank + extra_dims
229 def test_random(self, rank, extra_dims, np_type): argument
231 dims = rank + extra_dims
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | random_grad.py | 203 extra_dims = math_ops.range(
206 grad_mean = math_ops.reduce_sum(grad * dmean, axis=extra_dims)
207 grad_stddev = math_ops.reduce_sum(grad * dstddev, axis=extra_dims)
208 grad_minval = math_ops.reduce_sum(grad * dminval, axis=extra_dims)
209 grad_maxval = math_ops.reduce_sum(grad * dmaxval, axis=extra_dims)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2744 std::vector<int> extra_dims; in DynamicExpandDims() local
2747 extra_dims.push_back(-1); in DynamicExpandDims()
2749 extra_dims.push_back(1); in DynamicExpandDims()
2753 return DynamicReshape(input, slices, params, output, extra_dims, op_instance); in DynamicExpandDims()
|