Home
last modified time | relevance | path

Searched refs:inner_size (Results 1 – 15 of 15) 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()
Dreference_ops.h1880 int inner_size = 1; in Gather() local
1882 inner_size *= input_shape.Dims(i); in Gather()
1890 output_data + (outer * coords_count + i) * inner_size, in Gather()
1891 input_data + (outer * axis_size + coords_data[i]) * inner_size, in Gather()
1892 sizeof(T) * inner_size); in Gather()
2573 const int64_t inner_size = in RankOneSelect() local
2579 memcpy(output_data + offset, input_data + offset, inner_size * sizeof(T)); in RankOneSelect()
2580 offset += inner_size; in RankOneSelect()
/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc136 int64 inner_size = 1; in Compute() local
151 inner_size *= params.dim_size(i); in Compute()
162 {batch_size, outer_size, gather_dim_size, inner_size}); in Compute()
164 {batch_size, outer_size, N / batch_size, inner_size}); in Compute()
170 params.shaped<T, 3>({outer_size, gather_dim_size, inner_size}); in Compute()
171 auto out_flat = out->shaped<T, 3>({outer_size, N, inner_size}); in Compute()
Dreverse_op.cc51 const int64 inner_size = in ReverseRows() local
54 const int64 row_size = inner_size * middle_size; in ReverseRows()
55 DCHECK_EQ(input.dim_size(2), inner_size); in ReverseRows()
67 out_ptr -= inner_size; in ReverseRows()
68 memcpy(out_ptr, in_ptr, inner_size * sizeof(T)); in ReverseRows()
69 in_ptr += inner_size; in ReverseRows()
Dresource_variable_ops.cc703 int64 inner_size = 1; in Compute() local
705 inner_size *= params.dim_size(i); in Compute()
707 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute()
/external/python/cpython3/Modules/_blake2/clinic/
Dblake2s_impl.c.h18 int inner_size, int last_node);
40 int inner_size = 0; in py_blake2s_new() local
170 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2s_new()
171 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2s_new()
183 …e, &key, &salt, &person, fanout, depth, leaf_size, node_offset, node_depth, inner_size, last_node); in py_blake2s_new()
Dblake2b_impl.c.h18 int inner_size, int last_node);
40 int inner_size = 0; in py_blake2b_new() local
170 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2b_new()
171 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2b_new()
183 …e, &key, &salt, &person, fanout, depth, leaf_size, node_offset, node_depth, inner_size, last_node); in py_blake2b_new()
/external/python/cpython3/Modules/_blake2/
Dblake2s_impl.c93 int inner_size, int last_node) 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) 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/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dtranspose_conv.cc286 short2 inner_size; member
345 params3x3->inner_size.x = in Init3x3()
347 params3x3->inner_size.y = in Init3x3()
1175 if (params3x3.inner_size.x * params3x3.inner_size.y == 0) { in ConvolutionTransposed3x3()
1178 int groups_x = IntegralDivideRoundUp(params3x3.inner_size.x, groups_size.x); in ConvolutionTransposed3x3()
1179 int groups_y = IntegralDivideRoundUp(params3x3.inner_size.y, groups_size.y); in ConvolutionTransposed3x3()
/external/tensorflow/tensorflow/lite/experimental/ruy/
Ddispatch.h139 int inner_size = packed->rows; in CreatePackedLayout() local
142 (inner_size * scalar.size) % 1024 ? inner_size : inner_size + 64; in CreatePackedLayout()
144 packed->stride = inner_size; in CreatePackedLayout()
/external/python/cpython3/Lib/test/
Dtest_hashlib.py590 constructor(inner_size=i)
591 self.assertRaises(ValueError, constructor, inner_size=-1)
592 self.assertRaises(ValueError, constructor, inner_size=digest_size+1)
619 inner_size=7,
681 inner_size=7,
724 inner_size=7,
/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/Doc/library/
Dhashlib.rst311 node_depth=0, inner_size=0, last_node=False)
315 node_depth=0, inner_size=0, last_node=False)
370 * *inner_size*: inner digest size (0 to 64 for BLAKE2b, 0 to 32 for
663 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
667 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
671 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc507 const int64_t inner_size = dim_size * base_inner_size; in ConstFoldConcatenateOpDense() local
510 auto input_iter = input_attrs.begin() + outer * inner_size; in ConstFoldConcatenateOpDense()
511 for (int64_t inner = 0; inner < inner_size; ++inner) in ConstFoldConcatenateOpDense()