| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| cross_op.cc | 38 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); variable
39 OP_REQUIRES(ctx, inner_dim == 3,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| cross_op.cc | 60 auto inner_dim = in0.dim_size(in0.dims() - 1); variable
61 OP_REQUIRES(context, inner_dim == 3,
|
| training_ops.cc | 938 int64 inner_dim = 1; variable
1278 int64 inner_dim = 1; variable
1429 int64 inner_dim = 1; variable
1718 int64 inner_dim = 1; variable
2070 int64 inner_dim = 1; variable
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fft_ops_test.py | 313 inner_dim = size // 2 + 1
317 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
318 10).reshape((size,) * (dims - 1) + (inner_dim,))
327 inner_dim = size // 2 + 1
331 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
332 10).reshape((size,) * (dims - 1) + (inner_dim,))
340 inner_dim = size // 2 + 1
347 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
348 10).reshape((size,) * (dims - 1) + (inner_dim,))
360 inner_dim = size // 2 +
[all...] |
| segment_reduction_ops_test.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| spectral_grad.py | 126 inner_dim = input_shape[-1]
131 [inner_dim]
132 ], 0)) * _YMMask(inner_dim)
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorConvolution.h | 875 const int inner_dim = ((maxSharedMem \/ (sizeof(Scalar)) - kernel_size + 1 + 31) \/ 32) * 32; local
883 const int inner_dim = maxSharedMem \/ ((warpSize + kernel_size) * sizeof(Scalar)); local
938 int inner_dim = ((static_cast<int>(scaling_factor * kernel_size_x) - kernel_size_x + 1 + 32) \/ 32) * 32; local
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| math_ops.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers.py | [all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | [all...] |
