Home
last modified time | relevance | path

Searched refs:pad_value (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dpad.cc151 #define TF_LITE_PAD(type, op_name, scalar, pad_value) \ in Eval() argument
152 const scalar pad_value_copy = pad_value; \ in Eval()
160 float pad_value = op_context.constant_values == nullptr in Eval() local
165 TF_LITE_PAD(reference_ops, PadImageStyle, float, pad_value); in Eval()
167 TF_LITE_PAD(reference_ops, Pad, float, pad_value); in Eval()
171 TF_LITE_PAD(optimized_ops, PadImageStyle, float, pad_value); in Eval()
173 TF_LITE_PAD(optimized_ops, Pad, float, pad_value); in Eval()
178 uint8_t pad_value; in Eval() local
186 pad_value = static_cast<uint8_t>(op_context.output->params.zero_point); in Eval()
194 pad_value = *GetTensorData<uint8_t>(op_context.constant_values); in Eval()
[all …]
Dspace_to_batch_nd.cc118 #define TF_LITE_SPACE_TO_BATCH_ND(type, scalar, pad_value) \ in Eval() argument
120 op_params.output_offset = pad_value; \ in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dpad.cc122 #define TF_LITE_PAD(type, op_name, scalar, pad_value) \ in Eval() argument
123 const scalar pad_value_copy = pad_value; \ in Eval()
131 float pad_value = op_context.constant_values == nullptr in Eval() local
135 TF_LITE_PAD(reference_ops, PadImageStyle, float, pad_value); in Eval()
137 TF_LITE_PAD(reference_ops, Pad, float, pad_value); in Eval()
141 uint8_t pad_value; in Eval() local
149 pad_value = static_cast<uint8_t>(op_context.output->params.zero_point); in Eval()
158 pad_value = *GetTensorData<uint8_t>(op_context.constant_values); in Eval()
161 TF_LITE_PAD(reference_ops, PadImageStyle, uint8_t, pad_value); in Eval()
163 TF_LITE_PAD(reference_ops, Pad, uint8_t, pad_value); in Eval()
[all …]
Dpad_test.cc139 const float pad_value, const int* output_dims_data, in TestPadV2Float() argument
154 CreateFloatTensor(&pad_value, pad_value_dims, "pad value tensor"), in TestPadV2Float()
202 const int* pad_data, const float pad_value, in TestPadV2Quantized() argument
223 CreateQuantizedTensor(&pad_value, &pad_value_quantized, pad_value_dims, in TestPadV2Quantized()
286 const float pad_value = 42; in TF_LITE_MICRO_TEST() local
293 pad_values, pad_value, output_dims, golden, in TF_LITE_MICRO_TEST()
326 const float pad_value = 42; in TF_LITE_MICRO_TEST() local
340 pad_dims, pad_values, pad_value, pad_value_scale, pad_value_zero_point, in TF_LITE_MICRO_TEST()
374 const float pad_value = 42; in TF_LITE_MICRO_TEST() local
388 pad_dims, pad_values, pad_value, pad_value_scale, pad_value_zero_point, in TF_LITE_MICRO_TEST()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dpad_op.cc70 T pad_value = T(); in Compute() local
77 pad_value = context->input(2).scalar<T>()(); in Compute()
124 pad_value, &collapsed_output); in Compute()
133 OperateWithVariableRank(context, dims, in0, paddings, pad_value, output); in Compute()
209 T pad_value, Tensor* output) { in OperateWithVariableRank() argument
213 Operate<0>(context, input.tensor<T, 0>(), paddings, pad_value, output); in OperateWithVariableRank()
218 Operate<1>(context, input.flat<T>(), paddings, pad_value, output); in OperateWithVariableRank()
221 Operate<2>(context, input.tensor<T, 2>(), paddings, pad_value, output); in OperateWithVariableRank()
224 Operate<3>(context, input.tensor<T, 3>(), paddings, pad_value, output); in OperateWithVariableRank()
227 Operate<4>(context, input.tensor<T, 4>(), paddings, pad_value, output); in OperateWithVariableRank()
[all …]
Dpad_op.h35 T pad_value) { in operator()
38 To32Bit(output).device(d) = To32Bit(input).pad(paddings, pad_value); in operator()
40 output.device(d) = input.pad(paddings, pad_value); in operator()
/external/tensorflow/tensorflow/python/ops/signal/
Dshape_ops.py56 def frame(signal, frame_length, frame_step, pad_end=False, pad_value=0, axis=-1, argument
106 pad_value]):
133 pad_value = ops.convert_to_tensor(pad_value, signal.dtype)
134 pad_value.shape.assert_has_rank(0)
150 signal = array_ops.pad(signal, paddings, constant_values=pad_value)
/external/tensorflow/tensorflow/python/kernel_tests/signal/
Dshape_ops_test.py88 shape_ops.frame([1], 1, 1, pad_end=True, pad_value=[1])
96 signal, frame_length, frame_step, pad_end=True, pad_value=99))
112 pad_end=True, pad_value=99)
126 pad_end=True, pad_value=99, axis=1)
141 pad_end=True, pad_value=99, axis=3)
157 pad_end=pad_end, pad_value=99)
171 pad_end=True, pad_value=99)
196 pad_end=True, pad_value=99)
226 pad_end=True, pad_value=99)
284 pad_end=True, pad_value=99)
[all …]
/external/XNNPACK/src/
Dchannel-pad-nc.c25 const void* pad_value, in xnn_create_channel_pad_nc_x32() argument
76 channel_pad_op->pad_value = *((const uint32_t*) pad_value); in xnn_create_channel_pad_nc_x32()
126 .c = channel_pad_op->pad_value, in xnn_setup_channel_pad_nc_x32()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc168 const float pad_value = 1.5f; in TEST_P() local
171 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), in TEST_P()
175 expected->Fill(pad_value); in TEST_P()
208 const float pad_value = -5.123f; in TEST_P() local
214 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config); in TEST_P()
217 expected_array.Fill(pad_value); in TEST_P()
250 const float pad_value = -5.123f; in XLA_TEST_P() local
252 input_array.Fill(pad_value); in XLA_TEST_P()
260 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config); in XLA_TEST_P()
263 expected_array.Fill(pad_value); in XLA_TEST_P()
/external/XNNPACK/test/
Dchannel-pad-operator-tester.h109 const uint32_t pad_value = u32rng(); in TestX32() local
124 &pad_value, in TestX32()
145 ASSERT_EQ(pad_value, in TestX32()
153 ASSERT_EQ(pad_value, in TestX32()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dpad.h84 const T pad_value = *pad_value_ptr; in PadImpl() local
100 *out_ptr++ = pad_value; in PadImpl()
Dlegacy_reference_ops.h2047 const int32_t pad_value) { in SpaceToBatchND() argument
2049 op_params.output_offset = pad_value; in SpaceToBatchND()
2090 const Dims<4>& output_dims, const T pad_value) { in PadV2() argument
2101 const T pad_value_copy = pad_value; in PadV2()
2112 const Dims<4>& output_dims, const int32_t pad_value) { in Pad() argument
2113 const T converted_pad_value = static_cast<T>(pad_value); in Pad()
2124 const T pad_value = static_cast<T>(0); in Pad() local
2126 output_dims, pad_value); in Pad()
/external/angle/third_party/spirv-tools/src/source/opt/
Dfold_spec_constant_op_and_composite_pass.cpp330 uint32_t pad_value = 0; in EncodeIntegerAsWords() local
336 pad_value = all_ones; in EncodeIntegerAsWords()
367 words.push_back(pad_value); in EncodeIntegerAsWords()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfold_spec_constant_op_and_composite_pass.cpp330 uint32_t pad_value = 0; in EncodeIntegerAsWords() local
336 pad_value = all_ones; in EncodeIntegerAsWords()
367 words.push_back(pad_value); in EncodeIntegerAsWords()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfold_spec_constant_op_and_composite_pass.cpp330 uint32_t pad_value = 0; in EncodeIntegerAsWords() local
336 pad_value = all_ones; in EncodeIntegerAsWords()
367 words.push_back(pad_value); in EncodeIntegerAsWords()
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dimage_preprocessing_stage.cc188 int pad_value = params.padding_value(); in Pad() local
210 &pad_value, output_shape, output_data->data()); in Pad()
/external/XNNPACK/src/xnnpack/
Doperator.h193 uint32_t pad_value; member
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.signal.pbtxt25 …argspec: "args=[\'signal\', \'frame_length\', \'frame_step\', \'pad_end\', \'pad_value\', \'axis\'…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.signal.pbtxt25 …argspec: "args=[\'signal\', \'frame_length\', \'frame_step\', \'pad_end\', \'pad_value\', \'axis\'…
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3406 absl::string_view expected_conv_window, float pad_value) in ConvPaddingTestcase()
3410 pad_value(pad_value) {} in ConvPaddingTestcase()
3416 padding, orig_conv_window, expected_conv_window, pad_value); in ToString()
3422 float pad_value; member
3487 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
3488 LiteralUtil::CreateR0(testcase.pad_value))); in TEST_P()
3493 ShapeInference::InferPadShape(input->shape(), pad_value->shape(), in TEST_P()
3496 input, pad_value, padding_config)); in TEST_P()
3593 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
3594 LiteralUtil::CreateR0(testcase.pad_value))); in TEST_P()
[all …]
Dhlo_evaluator_test.cc654 auto pad_value = LiteralUtil::CreateR0<int32>(kPadValue); in TEST_F() local
656 b.AddInstruction(HloInstruction::CreateConstant(std::move(pad_value))); in TEST_F()
680 auto pad_value = LiteralUtil::CreateR0<float>(kPadValue); in TEST_P() local
682 b.AddInstruction(HloInstruction::CreateConstant(std::move(pad_value))); in TEST_P()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h280 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape() argument
288 SetDim(i, pad_value); in RuntimeShape()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5058 const T pad_value = *pad_value_ptr; in PadImpl() local
5062 output_data, pad_value, in PadImpl()
5069 pad_value, left_h_padding * output_width * output_depth); in PadImpl()
5076 pad_value, left_w_padding * output_depth); in PadImpl()
5083 pad_value, left_d_padding); in PadImpl()
5097 pad_value, right_d_padding); in PadImpl()
5103 pad_value, right_w_padding * output_depth); in PadImpl()
5110 pad_value, right_h_padding * output_width * output_depth); in PadImpl()
5117 pad_value, in PadImpl()
5215 const T pad_value = *pad_value_ptr; in PadImageStyleMemset() local
[all …]
/external/XNNPACK/include/
Dxnnpack.h940 const void* pad_value,

12