| /external/XNNPACK/test/ |
| D | static-constant-pad.cc | 35 float padding_value = f32dist(rng); in TEST_F() local
36 uint32_t quantized_padding_value = xnn_qs8_quantize(padding_value, scale, zero_point); in TEST_F()
61 …subgraph, pre_paddings.data(), post_paddings.data(), padding_value, input_id, output_id, /*flags=*… in TEST_F()
71 ASSERT_EQ(node->params.static_pad.padding_value, quantized_padding_value); in TEST_F()
87 float padding_value = f32dist(rng); in TEST_F() local
88 uint32_t quantized_padding_value = xnn_qu8_quantize(padding_value, scale, zero_point); in TEST_F()
113 …subgraph, pre_paddings.data(), post_paddings.data(), padding_value, input_id, output_id, /*flags=*… in TEST_F()
123 ASSERT_EQ(node->params.static_pad.padding_value, quantized_padding_value); in TEST_F()
137 float padding_value = f32dist(rng); in TEST_F() local
138 uint32_t padding_value_as_bits = float_as_uint32(padding_value); in TEST_F()
[all …]
|
| D | constant-pad-operator-tester.h | 146 const uint8_t padding_value = u8dist(rng); in TestX8() local
149 std::fill(output_ref.begin(), output_ref.end(), padding_value); in TestX8()
180 &padding_value, 0, &pad_op)); in TestX8()
210 << ", padding value = " << padding_value; in TestX8()
261 const uint16_t padding_value = u16dist(rng); in TestX16() local
264 std::fill(output_ref.begin(), output_ref.end(), padding_value); in TestX16()
295 &padding_value, 0, &pad_op)); in TestX16()
325 << ", padding value = " << padding_value; in TestX16()
376 const uint32_t padding_value = u32dist(rng); in TestX32() local
379 std::fill(output_ref.begin(), output_ref.end(), padding_value); in TestX32()
[all …]
|
| D | fusion.cc | 386 float padding_value = 0.0f; in TEST() local
394 .AddConstantPad(pre_paddings, post_paddings, padding_value, input_id, intermediate_id) in TEST()
432 float padding_value = 0.0f; in TEST() local
440 .AddConstantPad(pre_paddings, post_paddings, padding_value, input_id, intermediate_id) in TEST()
465 float padding_value = 1.0f; in TEST() local
473 .AddConstantPad(pre_paddings, post_paddings, padding_value, input_id, intermediate_id) in TEST()
498 float padding_value = 0.0f; in TEST() local
505 .AddConstantPad(pre_paddings, post_paddings, padding_value, input_id, intermediate_id) in TEST()
541 float padding_value = 0.0f; in TEST() local
548 .AddConstantPad(pre_paddings, post_paddings, padding_value, input_id, intermediate_id) in TEST()
[all …]
|
| /external/XNNPACK/src/subgraph/ |
| D | static-constant-pad.c | 45 &node->params.static_pad.padding_value, in create_constant_pad_operator()
52 &node->params.static_pad.padding_value, in create_constant_pad_operator()
64 &node->params.static_pad.padding_value, in create_constant_pad_operator()
151 float padding_value, in xnn_define_static_constant_pad() argument
262 node->params.static_pad.padding_value = float_as_uint32(padding_value); in xnn_define_static_constant_pad()
269 …node->params.static_pad.padding_value = xnn_qs8_quantize(padding_value, output_scale, output_zero_… in xnn_define_static_constant_pad()
278 …node->params.static_pad.padding_value = xnn_qu8_quantize(padding_value, output_scale, output_zero_… in xnn_define_static_constant_pad()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op_gpu.cu.cc | 48 const int upper_diag_index, const T padding_value, in MatrixDiagKernel() argument
67 output_ptr[index] = padding_value; in MatrixDiagKernel()
79 const Eigen::Index max_diag_len, const T padding_value, in Compute()
96 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
105 const int upper_diag_index, const T padding_value, in MatrixDiagPartKernel() argument
128 output_ptr[index] = padding_value; in MatrixDiagPartKernel()
140 const Eigen::Index max_diag_len, const T padding_value, in Compute()
157 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
|
| D | matrix_diag_op.cc | 65 T padding_value(0); in Compute() local
95 padding_value = padding_in.flat<T>()(0); in Compute()
146 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
177 T padding_value(0); in Compute() local
219 padding_value = padding_value_tensor.flat<T>()(0); in Compute()
281 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
358 const Eigen::Index max_diag_len, const T padding_value, in Compute()
370 &padding_value, &left_align_superdiagonal, in Compute()
393 output(batch, i, j) = padding_value; in Compute()
414 const Eigen::Index max_diag_len, const T padding_value, in Compute()
[all …]
|
| D | matrix_diag_op.h | 54 const T padding_value, const bool left_align_superdiagonal,
65 const Eigen::Index max_diag_len, const T padding_value,
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 21 uint32_t padding_value, in create_constant_pad_nd() argument
46 constant_pad_op->pad_value = padding_value; in create_constant_pad_nd()
62 const void* padding_value, in xnn_create_constant_pad_nd_x8() argument
66 const uint32_t padding_pattern = *((const uint8_t*) padding_value); in xnn_create_constant_pad_nd_x8()
72 const void* padding_value, in xnn_create_constant_pad_nd_x16() argument
76 const uint32_t padding_pattern = *((const uint16_t*) padding_value); in xnn_create_constant_pad_nd_x16()
82 const void* padding_value, in xnn_create_constant_pad_nd_x32() argument
87 …*((const uint32_t*) padding_value), flags, xnn_operator_type_constant_pad_nd_x32, constant_pad_op_… in xnn_create_constant_pad_nd_x32()
171 .padding_value = constant_pad_op->pad_value, in setup_constant_pad_nd()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | diag_op_test.py | 416 for padding_value, align in zip_to_first_list_length([0, 555, -11],
423 padding_value=padding_value,
426 solution = (solution + padding_value * mask).astype(dtype)
479 for padding_value, align in zip_to_first_list_length([0, 555, -11],
489 padding_value=padding_value,
492 solution = solution + padding_value * mask
508 padding_value=padding_value,
511 solution = solution + padding_value * mask
527 padding_value=padding_value,
530 solution = solution + padding_value * mask
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagPartV3.pbtxt | 17 name: "padding_value"
56 padding_value ; otherwise.
66 padding_value ; otherwise.
130 # padding_value = 9
131 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagPartV2.pbtxt | 17 name: "padding_value"
44 padding_value ; otherwise.
54 padding_value ; otherwise.
89 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
|
| D | api_def_MatrixDiagV2.pbtxt | 33 name: "padding_value"
68 padding_value ; otherwise
77 padding_value ; otherwise
129 # Rectangular matrix with inferred num_cols and padding_value = 9.
130 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| D | api_def_MatrixDiagV3.pbtxt | 33 name: "padding_value"
80 padding_value ; otherwise
89 padding_value ; otherwise
169 # Rectangular matrix with inferred num_cols and padding_value = 9.
170 tf.matrix_diag(diagonal, k = -1, num_rows = 3, padding_value = 9)
|
| /external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| D | rnn.h | 249 double padding_value = 0.0,
268 padding_value,
312 double padding_value = 0,
314 return at::pad_sequence(sequences, batch_first, padding_value, padding_side);
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorTransformerFunctions.cu | 251 T padding_value, in add_padding_1() argument
270 output[output_offset] = padding_value; in add_padding_1()
280 output[output_offset] = padding_value; in add_padding_1()
289 T padding_value, in add_padding_2() argument
312 output[output_offset + i] = padding_value; in add_padding_2()
324 output[output_offset + i] = padding_value; in add_padding_2()
333 T padding_value, in add_padding_3() argument
361 output[output_offset + i] = padding_value; in add_padding_3()
376 output[output_offset + i] = padding_value; in add_padding_3()
385 T padding_value, in add_padding_kernelLauncher() argument
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 46 for (int64_t padding_value : explicit_paddings) { in CheckValidPadding() local
47 if (padding_value < 0) { in CheckValidPadding()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | PackedSequence.cpp | 142 …or& data, const Tensor& _batch_sizes, bool batch_first, const Scalar& padding_value, int64_t total… in _pad_packed_sequence() argument
166 auto output = at::full(output_size, padding_value, data.options()); in _pad_packed_sequence()
205 Tensor pad_sequence(TensorList sequences, bool batch_first, double padding_value, const c10::string… in pad_sequence() argument
228 Tensor out = at::full(out_dims, padding_value, sequences[0].options()); in pad_sequence()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorImagePatch.h | 101 const Index num_coeff_to_pad, const Scalar padding_value,
104 const Packet padded_packet = internal::pset1<Packet>(padding_value);
112 dst_data[dst_index + i] = padding_value;
134 … PaddingType padding_type, Scalar padding_value)
140 … m_padding_type(padding_type), m_padding_value(padding_value) {}
148 Scalar padding_value)
155 … m_padding_type(PADDING_VALID), m_padding_value(padding_value) {}
187 Scalar padding_value() const { return m_padding_value; }
250 m_paddingValue = op.padding_value();
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 516 for padding_value, align in zip_to_first_list_length([555, -11],
521 solution = solution + (mask * padding_value)
528 "padding_value": padding_value,
756 for padding_value, align in zip_to_first_list_length([555, -11],
761 solution = solution + (mask * padding_value)
766 "padding_value": padding_value,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 263 xla::XlaOp padding_value = zero; in Compile() local
272 padding_value = context->Input(4); in Compile()
318 xla::XlaOp output = xla::Broadcast(padding_value, output_shape.dim_sizes()); in Compile()
374 xla::XlaOp padding_value = zero; in Compile() local
381 padding_value = context->Input(2); in Compile()
431 single_diag = xla::Pad(single_diag, padding_value, padding_config); in Compile()
|
| /external/pytorch/torch/_inductor/ |
| D | jagged_lowerings.py | 114 padding_value: float = 0.0,
138 padding_value,
170 padding_value,
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-004.md | 14 padding_value = context->input(4).flat<T>()(0);
25 tf.raw_ops.MatrixDiagV2(diagonal=d, k=0, num_rows=0, num_cols=0, padding_value=p)
|
| /external/libvpx/test/ |
| D | buffer.h | 93 void SetPadding(const T padding_value);
278 void Buffer<T>::SetPadding(const T padding_value) { in SetPadding() argument
280 padding_value_ = padding_value; in SetPadding()
284 src[i] = padding_value; in SetPadding()
|
| /external/pytorch/torch/nn/utils/ |
| D | rnn.py | 345 padding_value: float = 0.0,
405 sequence.data, sequence.batch_sizes, batch_first, padding_value, max_seq_length
421 padding_value: float = 0.0,
479 sequences, batch_first, padding_value, padding_side # type: ignore[arg-type]
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | encoder_common.cc | 114 const int32_t padding_value, in CopyDataToTensorAndPadOrTruncate() argument
126 output_buffer[output_offset] = padding_value; in CopyDataToTensorAndPadOrTruncate()
|