Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc91 int64 inner_size = 1; in Compute() local
99 inner_size *= params.dim_size(i); in Compute()
104 if (N > 0 && outer_size > 0 && inner_size > 0) { in Compute()
106 params.shaped<T, 3>({outer_size, gather_dim_size, inner_size}); in Compute()
108 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.cc685 int64 inner_size = 1; in Compute() local
687 inner_size *= params.dim_size(i); in Compute()
689 auto params_flat = params.shaped<T, 3>({1, gather_dim_size, inner_size}); in Compute()
/external/python/cpython3/Modules/_blake2/clinic/
Dblake2b_impl.c.h18 int inner_size, int last_node);
36 int inner_size = 0; in py_blake2b_new() local
40 …person, &fanout, &depth, &leaf_size_obj, &node_offset_obj, &node_depth, &inner_size, &last_node)) { in py_blake2b_new()
43 … &salt, &person, fanout, depth, leaf_size_obj, node_offset_obj, node_depth, inner_size, last_node); in py_blake2b_new()
Dblake2s_impl.c.h18 int inner_size, int last_node);
36 int inner_size = 0; in py_blake2s_new() local
40 …person, &fanout, &depth, &leaf_size_obj, &node_offset_obj, &node_depth, &inner_size, &last_node)) { in py_blake2s_new()
43 … &salt, &person, fanout, depth, leaf_size_obj, node_offset_obj, node_depth, inner_size, last_node); in py_blake2s_new()
/external/python/cpython3/Modules/_blake2/
Dblake2b_impl.c93 int inner_size, int last_node) in py_blake2b_new_impl() argument
194 if (inner_size < 0 || inner_size > BLAKE2B_OUTBYTES) { in py_blake2b_new_impl()
200 self->param.inner_length = inner_size; in py_blake2b_new_impl()
Dblake2s_impl.c93 int inner_size, int last_node) in py_blake2s_new_impl() argument
194 if (inner_size < 0 || inner_size > BLAKE2S_OUTBYTES) { in py_blake2s_new_impl()
200 self->param.inner_length = inner_size; in py_blake2s_new_impl()
/external/python/cpython3/Lib/test/
Dtest_hashlib.py571 constructor(inner_size=i)
572 self.assertRaises(ValueError, constructor, inner_size=-1)
573 self.assertRaises(ValueError, constructor, inner_size=digest_size+1)
600 inner_size=7,
662 inner_size=7,
705 inner_size=7,
/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc226 const int inner_size = shape_common.Dims(3); in RunOneLogSoftmaxTest() local
229 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/lite/kernels/internal/reference/
Dreference_ops.h2955 int inner_size = 1; in Gather() local
2957 inner_size *= input_shape.Dims(i); in Gather()
2965 output_data + (outer * coords_count + i) * inner_size, in Gather()
2966 input_data + (outer * axis_size + coords_data[i]) * inner_size, in Gather()
2967 sizeof(T) * inner_size); in Gather()
3886 int inner_size = 1; in ArgMinMax() local
3890 inner_size *= input1_shape.Dims(i); in ArgMinMax()
3894 for (int inner = 0; inner < inner_size; ++inner) { in ArgMinMax()
3895 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
3899 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
[all …]