Home
last modified time | relevance | path

Searched refs:inner_size (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Darg_min_max.h43 int inner_size = 1; in ArgMinMax() local
47 inner_size *= input1_shape.Dims(i); in ArgMinMax()
50 for (int inner = 0; inner < inner_size; ++inner) { in ArgMinMax()
51 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
55 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
61 output_data[outer * inner_size + inner] = min_max_index; in ArgMinMax()
Dgather.h45 int inner_size = 1; in Gather() local
47 inner_size *= input_shape.Dims(i); in Gather()
56 output_data + (outer * coords_count + i) * inner_size, in Gather()
57 input_data + (outer * axis_size + coords_data[i]) * inner_size, in Gather()
58 sizeof(T) * inner_size); in Gather()
Dreference_ops.h1504 int64_t inner_size; in RankOneSelect() local
1506 inner_size = MatchingFlatSize(input_x_shape, input_y_shape, output_shape); in RankOneSelect()
1511 inner_size = in RankOneSelect()
1518 memcpy(output_data + offset, input_data + offset, inner_size * sizeof(T)); in RankOneSelect()
1519 offset += inner_size; in RankOneSelect()
/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc139 int64 inner_size = 1; in Compute() local
154 inner_size *= params.dim_size(i); in Compute()
160 if (inner_size == 0) return; in Compute()
166 {batch_size, outer_size, gather_dim_size, inner_size}); in Compute()
168 {batch_size, outer_size, N / batch_size, inner_size}); in Compute()
174 params.shaped<T, 3>({outer_size, gather_dim_size, inner_size}); in Compute()
175 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); in Compute()
Dreverse_op.cc48 const int64 inner_size = in ReverseRows() local
51 const int64 row_size = inner_size * middle_size; in ReverseRows()
52 DCHECK_EQ(input.dim_size(2), inner_size); in ReverseRows()
64 out_ptr -= inner_size; in ReverseRows()
65 memcpy(out_ptr, in_ptr, inner_size * sizeof(T)); in ReverseRows()
66 in_ptr += inner_size; in ReverseRows()
Dresource_variable_ops.cc689 int64 inner_size = 1; in Compute() local
691 inner_size *= params.dim_size(i); in Compute()
693 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute()
/external/python/cpython3/Modules/_blake2/clinic/
Dblake2s_impl.c.h19 int inner_size, int last_node, int usedforsecurity);
41 int inner_size = 0; in py_blake2s_new() local
172 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2s_new()
173 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2s_new()
194 …ey, &salt, &person, fanout, depth, leaf_size, node_offset, node_depth, inner_size, last_node, used… in py_blake2s_new()
Dblake2b_impl.c.h19 int inner_size, int last_node, int usedforsecurity);
41 int inner_size = 0; in py_blake2b_new() local
172 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2b_new()
173 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2b_new()
194 …ey, &salt, &person, fanout, depth, leaf_size, node_offset, node_depth, inner_size, last_node, used… in py_blake2b_new()
/external/python/cpython3/Modules/_blake2/
Dblake2s_impl.c93 int inner_size, int last_node, int usedforsecurity) in py_blake2s_new_impl() argument
179 if (inner_size < 0 || inner_size > BLAKE2S_OUTBYTES) { in py_blake2s_new_impl()
185 self->param.inner_length = inner_size; in py_blake2s_new_impl()
Dblake2b_impl.c93 int inner_size, int last_node, int usedforsecurity) in py_blake2b_new_impl() argument
179 if (inner_size < 0 || inner_size > BLAKE2B_OUTBYTES) { in py_blake2b_new_impl()
185 self->param.inner_length = inner_size; in py_blake2b_new_impl()
/external/ruy/ruy/
Dtrace.h390 int inner_size = in str() local
392 if (inner_size != layout.stride()) { in str()
403 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local
404 if (inner_size != layout.stride) { in str()
415 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str() local
416 if (inner_size != layout.stride) { in str()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc232 const int inner_size = shape_common.Dims(3); in RunOneLogSoftmaxTest() local
235 reverse_scaling_right_shift, diff_min, outer_size, inner_size, input_data, in RunOneLogSoftmaxTest()
/external/python/cpython3/Lib/test/
Dtest_hashlib.py644 constructor(inner_size=i)
645 self.assertRaises(ValueError, constructor, inner_size=-1)
646 self.assertRaises(ValueError, constructor, inner_size=digest_size+1)
673 inner_size=7,
735 inner_size=7,
778 inner_size=7,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_helpers.inc307 // `inner_size` allows to shuffle array attributes created from rank 2 tensors
310 int inner_size = 1) {
313 assert(attr.size() % inner_size == 0);
314 assert(attr.size() / inner_size == permutation.size());
320 for (size_t j = 0; j < inner_size; ++j) {
321 shuffled[i * inner_size + j] = values[permutation[i] * inner_size + j];
/external/python/cpython3/Doc/library/
Dhashlib.rst320 node_depth=0, inner_size=0, last_node=False, \
325 node_depth=0, inner_size=0, last_node=False, \
381 * *inner_size*: inner digest size (0 to 64 for BLAKE2b, 0 to 32 for
674 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
678 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
682 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc721 const int64_t inner_size = dim_size * base_inner_size; in ConstFoldConcatenateOpDense() local
724 auto input_iter = input_attrs.begin() + outer * inner_size; in ConstFoldConcatenateOpDense()
725 for (int64_t inner = 0; inner < inner_size; ++inner) in ConstFoldConcatenateOpDense()