| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 51 Eigen::Index inner_dim = 1, outer_dim = 1; in operator() local
55 inner_dim *= input_dims[i]; in operator()
71 if (inner_dim > num_threads * 32) { in operator()
76 const Eigen::Index inner_block_size = Eigen::divup(inner_dim, num_blocks); in operator()
81 {inner_dim}); in operator()
93 const auto compute = [inner_dim, outer_dim, num_blocks, inner_block_size, in operator()
99 inner_dim_limit = std::min(inner_dim, inner_dim_limit); in operator()
106 auto in = Input(my_job_start + i * inner_dim, my_job_len); in operator()
128 const Eigen::Index parallel_cell_size = inner_dim; in operator()
129 const Eigen::Index total_workload = outer_dim * inner_dim; in operator()
[all …]
|
| D | cross_op.cc | 60 auto inner_dim = in0.dim_size(in0.dims() - 1); in Compute() local
61 OP_REQUIRES(context, inner_dim == 3, in Compute()
|
| D | training_ops.cc | 172 typename TTypes<Tindex>::ConstVec indices, int64 inner_dim, in operator ()()
178 const int in_bytes = inner_dim * sizeof(T) * 3; in operator ()()
179 const int out_bytes = inner_dim * sizeof(T) * 2; in operator ()()
180 const int cycles = inner_dim * (Eigen::TensorOpCost::AddCost<T>() * 2 + in operator ()()
184 if (inner_dim > 1) { in operator ()()
282 int64 inner_dim) { in operator ()()
289 if (inner_dim > 1) { in operator ()()
600 int64 inner_dim, bool multiply_linear_by_lr) { in operator ()()
611 if (inner_dim > 1) { in operator ()()
1424 int64 inner_dim = 1; in Compute() local
[all …]
|
| D | training_ops.h | 103 typename TTypes<Tindex>::ConstVec indices, int64 inner_dim,
125 typename TTypes<Tindex>::ConstVec indices, int64 inner_dim);
190 int64 inner_dim, bool multiply_linear_by_lr);
|
| D | count_ops.cc | 41 int inner_dim = is_1d ? 1 : 2; in OutputSparse() local
43 0, TensorShape({total_values, inner_dim}), &indices)); in OutputSparse()
|
| D | training_ops_gpu.cu.cc | 510 typename TTypes<Tindex>::ConstVec indices, int64 inner_dim, in operator ()()
572 int64 inner_dim) { in operator ()()
781 typename TTypes<Tindex>::ConstVec indices, int64 inner_dim, in operator ()()
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 369 inner_dim = size // 2 + 1
374 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
375 10).reshape((size,) * (dims - 1) + (inner_dim,))
386 inner_dim = size // 2 + 1
391 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
392 10).reshape((size,) * (dims - 1) + (inner_dim,))
404 inner_dim = size // 2 + 1
412 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
413 10).reshape((size,) * (dims - 1) + (inner_dim,))
427 inner_dim = size // 2 + 1
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cross_op.cc | 39 auto inner_dim = in0_shape.dim_size(in0_shape.dims() - 1); in Compile() local
40 OP_REQUIRES(ctx, inner_dim == 3, in Compile()
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 311 inner_dim = input_shape[-1]
316 [inner_dim]
317 ], 0)) * _ymask(inner_dim)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 975 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
977 const = np.random.randint(5, size=(outer_dim, inner_dim))
991 [name, outer_dim, ratio, inner_dim,
998 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
1001 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
1006 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
1009 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
|
| /external/libaom/libaom/av1/encoder/ |
| D | mathutils.h | 94 const int m1_rows, const int inner_dim, in multiply_mat() argument
102 for (inner = 0; inner < inner_dim; ++inner) in multiply_mat()
103 sum += m1[row * inner_dim + inner] * m2[inner * m2_cols + col]; in multiply_mat()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 875 … const int inner_dim = ((maxSharedMem / (sizeof(Scalar)) - kernel_size + 1 + 31) / 32) * 32;
876 maxX = numext::mini<int>(inner_dim, numX);
883 const int inner_dim = maxSharedMem / ((warpSize + kernel_size) * sizeof(Scalar));
884 const int maxP = numext::mini<int>(inner_dim, numP);
885 … maxX = numext::mini<int>(maxSharedMem / (inner_dim * sizeof(Scalar)) - kernel_size + 1, numX);
938 …int inner_dim = ((static_cast<int>(scaling_factor * kernel_size_x) - kernel_size_x + 1 + 32) / 32)…
939 const int maxX = numext::mini<int>(inner_dim, numX);
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 469 inner_dim = min(parent_rank, current_field_rank - 1)
470 if inner_dim <= new_parent_rank:
472 return _merge_dims_generic(current_field, new_parent_rank, inner_dim)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 1942 auto inner_dim = c->Dim(input, rank - 1); in __anon57f5a5dd1702() local
1944 TF_RETURN_IF_ERROR(c->Divide(inner_dim, 8, in __anon57f5a5dd1702()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1205 int64 inner_dim = RandomDim(); in TEST_F() local
1207 x_dims[ndims - 1] = inner_dim; in TEST_F()
1208 y_dims[ndims - 2] = inner_dim; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2506 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate() argument
2507 return product * output_shape.dimensions(inner_dim); in EmitFastConcatenate()
|