| /external/tensorflow/tensorflow/core/kernels/ |
| D | redux_functor.h | 43 int64 inner_dim = input_dims[num_dims - 1]; in operator() local
48 const int64 parallel_cell_size = inner_dim; in operator()
49 const int64 total_workload = outer_dim * inner_dim; in operator()
68 Tensor buffer(DataTypeToEnum<AccumT>::v(), {num_blocks, inner_dim}); in operator()
80 const auto compute = [inner_dim, num_blocks, outer_block_size, buffer_data, in operator()
88 Buffer buf(buffer_data + start * inner_dim, inner_dim); in operator()
90 auto in = Input(input_data + i * inner_dim, inner_dim); in operator()
98 const int64 compute_size = outer_block_size * inner_dim; in operator()
108 auto buf0 = Buffer(buffer_data, inner_dim); in operator()
111 auto buf = Buffer(buffer_data + i * inner_dim, inner_dim); in operator()
|
| 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 | 1019 int64 inner_dim = 1; in Compute() local
1024 inner_dim *= grad.dim_size(d); in Compute()
1031 OP_REQUIRES(ctx, inner_dim > 0, in Compute()
1036 if (inner_dim > 1) { in Compute()
1367 int64 inner_dim = 1; in Compute() local
1372 inner_dim *= grad.dim_size(d); in Compute()
1380 OP_REQUIRES(ctx, inner_dim > 0, in Compute()
1385 if (inner_dim > 1) { in Compute()
1525 int64 inner_dim = 1; in Compute() local
1530 inner_dim *= grad.dim_size(d); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 360 inner_dim = size // 2 + 1
364 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
365 10).reshape((size,) * (dims - 1) + (inner_dim,))
374 inner_dim = size // 2 + 1
378 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
379 10).reshape((size,) * (dims - 1) + (inner_dim,))
388 inner_dim = size // 2 + 1
395 c2r = np.mod(np.arange(np.power(size, dims - 1) * inner_dim),
396 10).reshape((size,) * (dims - 1) + (inner_dim,))
409 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 | 276 inner_dim = input_shape[-1]
281 [inner_dim]
282 ], 0)) * _ymask(inner_dim)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 927 def _runGraph(self, op_functor, outer_dim, ratio, inner_dim, dtype): argument
929 const = np.random.randint(5, size=(outer_dim, inner_dim))
943 [name, outer_dim, ratio, inner_dim,
950 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
953 self._runGraph(op_functor, outer_dim, ratio, inner_dim, dtype)
958 for outer_dim, ratio, inner_dim, dtype in itertools.product(*self.options):
961 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/core/ops/ |
| D | math_ops.cc | 1696 auto inner_dim = c->Dim(input, rank - 1); in __anonb22bfa861402() local
1698 TF_RETURN_IF_ERROR(c->Divide(inner_dim, 8, in __anonb22bfa861402()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 1663 for inner_dim in static_inner_dims:
1664 if inner_dim is None:
1667 flattened_dimension *= inner_dim
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1204 int64 inner_dim = RandomDim(); in TEST_F() local
1206 x_dims[ndims - 1] = inner_dim; in TEST_F()
1207 y_dims[ndims - 2] = inner_dim; in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2445 [&](int64 product, int64 inner_dim) { in EmitFastConcatenate() argument
2446 return product * output_shape.dimensions(inner_dim); in EmitFastConcatenate()
|