Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Darg_min_max.h54 int inner_size = 1; in ArgMinMax() local
58 inner_size *= input1_shape.Dims(i); in ArgMinMax()
61 for (int inner = 0; inner < inner_size; ++inner) { in ArgMinMax()
62 auto min_max_value = input1_data[outer * axis_size * inner_size + inner]; in ArgMinMax()
66 input1_data[(outer * axis_size + i) * inner_size + inner]; in ArgMinMax()
72 output_data[outer * inner_size + inner] = min_max_index; in ArgMinMax()
Dgather.h62 int inner_size = 1; in Gather() local
64 inner_size *= input_shape.Dims(i); in Gather()
80 (((batch * outer_size) + outer) * coord_size + i) * inner_size, in Gather()
83 inner_size, in Gather()
84 sizeof(T) * inner_size); in Gather()
Dselect.h58 int64_t inner_size; in RankOneSelect() local
60 inner_size = MatchingFlatSize(input_x_shape, input_y_shape, output_shape); in RankOneSelect()
65 inner_size = in RankOneSelect()
72 memcpy(output_data + offset, input_data + offset, inner_size * sizeof(T)); in RankOneSelect()
73 offset += inner_size; in RankOneSelect()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_layout_helper.cc43 int inner_size) { in ShuffleArrayAttr() argument
46 assert(attr.size() % inner_size == 0); in ShuffleArrayAttr()
47 assert(attr.size() / inner_size == permutation.size()); in ShuffleArrayAttr()
53 for (size_t j = 0; j < inner_size; ++j) { in ShuffleArrayAttr()
54 shuffled[i * inner_size + j] = values[permutation[i] * inner_size + j]; in ShuffleArrayAttr()
Dtf_ops_layout_helper.h42 int inner_size = 1);
/external/tensorflow/tensorflow/core/kernels/
Dgather_op.cc139 int64_t 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_t inner_size = in ReverseRows() local
51 const int64_t 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.cc754 int64_t inner_size = 1; in Compute() local
756 inner_size *= params.dim_size(i); in Compute()
758 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
147 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2s_new()
148 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2s_new()
169 …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
147 inner_size = _PyLong_AsInt(fastargs[10]); in py_blake2b_new()
148 if (inner_size == -1 && PyErr_Occurred()) { in py_blake2b_new()
169 …ey, &salt, &person, fanout, depth, leaf_size, node_offset, node_depth, inner_size, last_node, used… in py_blake2b_new()
/external/ComputeLibrary/tests/validation/reference/
DSelect.cpp64 const int inner_size = x.num_elements() / outer_size; in select() local
70 memcpy(output_ptr + offset, input_ptr + offset, inner_size * sizeof(T)); in select()
71 offset += inner_size; in select()
/external/ComputeLibrary/src/cpu/kernels/select/generic/neon/
Dimpl.cpp128 const int inner_size = (in1->info()->total_size() / in1->info()->element_size()) / outer_size; in select_op_not_same_rank() local
136 for(; x <= offset + inner_size - step; x += step) in select_op_not_same_rank()
140 if(x <= offset + inner_size - (step / 2)) in select_op_not_same_rank()
145 for(; x < offset + inner_size; ++x) in select_op_not_same_rank()
149 offset += inner_size; in select_op_not_same_rank()
/external/python/cpython3/Modules/_blake2/
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()
Dblake2s_impl.c92 int inner_size, int last_node, int usedforsecurity) in py_blake2s_new_impl() argument
178 if (inner_size < 0 || inner_size > BLAKE2S_OUTBYTES) { in py_blake2s_new_impl()
184 self->param.inner_length = inner_size; in py_blake2s_new_impl()
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference_beta.cc149 int32_t inner_size = 1; in GetReduceInnerSize() local
155 inner_size *= input_shape[index_value]; in GetReduceInnerSize()
157 return inner_size; in GetReduceInnerSize()
1202 const int32_t inner_size = ShapeSize(right_rank, right_shape); in OpInnerBroadcast() local
1203 const int32_t outer_size = output_size / inner_size; in OpInnerBroadcast()
1205 if (inner_size == 1) { in OpInnerBroadcast()
1207 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1208 auto output = MatrixMap<T>(output_values, inner_size, outer_size); in OpInnerBroadcast()
1211 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1212 auto right = ConstRowVectorMap<T>(right_values, inner_size); in OpInnerBroadcast()
[all …]
Donedevice_train_palm_detection_filter_inference.cc150 int32_t inner_size = 1; in GetReduceInnerSize() local
156 inner_size *= input_shape[index_value]; in GetReduceInnerSize()
158 return inner_size; in GetReduceInnerSize()
1203 const int32_t inner_size = ShapeSize(right_rank, right_shape); in OpInnerBroadcast() local
1204 const int32_t outer_size = output_size / inner_size; in OpInnerBroadcast()
1206 if (inner_size == 1) { in OpInnerBroadcast()
1208 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1209 auto output = MatrixMap<T>(output_values, inner_size, outer_size); in OpInnerBroadcast()
1212 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1213 auto right = ConstRowVectorMap<T>(right_values, inner_size); in OpInnerBroadcast()
[all …]
Donedevice_train_palm_detection_filter_inference_v2.cc149 int32_t inner_size = 1; in GetReduceInnerSize() local
155 inner_size *= input_shape[index_value]; in GetReduceInnerSize()
157 return inner_size; in GetReduceInnerSize()
1203 const int32_t inner_size = ShapeSize(right_rank, right_shape); in OpInnerBroadcast() local
1204 const int32_t outer_size = output_size / inner_size; in OpInnerBroadcast()
1206 if (inner_size == 1) { in OpInnerBroadcast()
1208 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1209 auto output = MatrixMap<T>(output_values, inner_size, outer_size); in OpInnerBroadcast()
1212 auto left = ConstMatrixMap<T>(left_values, inner_size, outer_size); in OpInnerBroadcast()
1213 auto right = ConstRowVectorMap<T>(right_values, inner_size); in OpInnerBroadcast()
[all …]
/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/python/kernel_tests/math_ops/
Dsegment_reduction_ops_test.py319 for inner_size in [1, 2, 3, 32]:
320 shape = indices.shape + (inner_size,)
587 for inner_size in [1, 2, 3, 32]:
588 shape = [n, inner_size]
927 for inner_size in (1, 2, 3, 32):
929 tf_ygrad, np_ygrad = self._input([3, inner_size],
948 for inner_size in (1, 2, 3, 32):
950 tf_ygrad, np_ygrad = self._input([3, inner_size],
/external/python/cpython3/Lib/test/
Dtest_hashlib.py647 constructor(inner_size=i)
648 self.assertRaises(ValueError, constructor, inner_size=-1)
649 self.assertRaises(ValueError, constructor, inner_size=digest_size+1)
676 inner_size=7,
738 inner_size=7,
781 inner_size=7,
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_c_api_client.cc476 int inner_size, xla::PjRtCApiClient* client) { in Convert2DCBuffersToCppBuffers() argument
480 output_list.reserve(inner_size); in Convert2DCBuffersToCppBuffers()
481 for (size_t j = 0; j < inner_size; ++j) { in Convert2DCBuffersToCppBuffers()
525 size_t inner_size = numoutputs_args.num_outputs; in Execute() local
529 c_buffer_lists_holder[i].resize(inner_size); in Execute()
/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/tensorflow/tensorflow/compiler/xla/pjrt/c/
Dpjrt_c_api_wrapper_impl.cc385 PJRT_Buffer*** c_lists, size_t outer_size, size_t inner_size) { in Convert2DCBuffersToCppBuffers() argument
390 cpp_list.reserve(inner_size); in Convert2DCBuffersToCppBuffers()
391 for (int j = 0; j < inner_size; ++j) { in Convert2DCBuffersToCppBuffers()
/external/python/cpython3/Doc/library/
Dhashlib.rst329 node_depth=0, inner_size=0, last_node=False, \
334 node_depth=0, inner_size=0, last_node=False, \
390 * *inner_size*: inner digest size (0 to 64 for BLAKE2b, 0 to 32 for
683 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
687 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
691 ... leaf_size=LEAF_SIZE, inner_size=INNER_SIZE,
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc854 const int64_t inner_size = dim_size * base_inner_size; in ConstFoldConcatenateOpDense() local
857 auto input_iter = input_attrs.begin() + outer * inner_size; in ConstFoldConcatenateOpDense()
858 for (int64_t inner = 0; inner < inner_size; ++inner) in ConstFoldConcatenateOpDense()

12