| /external/tensorflow/tensorflow/lite/kernels/ |
| pad.cc | 149 #define TF_LITE_PAD(type, op_name, scalar, pad_value) \
159 const scalar pad_value_copy = pad_value; \
167 float pad_value = op_context.constant_values == nullptr local
172 TF_LITE_PAD(reference_ops, PadImageStyle, float, pad_value);
174 TF_LITE_PAD(reference_ops, Pad, float, pad_value);
178 TF_LITE_PAD(optimized_ops, PadImageStyle, float, pad_value);
180 TF_LITE_PAD(optimized_ops, Pad, float, pad_value);
185 uint8_t pad_value; local
193 pad_value = static_cast<uint8_t>(op_context.output->params.zero_point);
201 pad_value = *GetTensorData<uint8_t>(op_context.constant_values)
218 int8_t pad_value; local
243 int32_t pad_value = local
254 int64_t pad_value = local
[all...] |
| space_to_batch_nd.cc | 116 #define TF_LITE_SPACE_TO_BATCH_ND(type, scalar, pad_value) \
118 op_params.output_offset = pad_value; \
|
| /external/tensorflow/tensorflow/core/kernels/ |
| pad_op.h | 30 // Pad "input" into "output", as specified by "paddings" and "pad_value".
35 T pad_value) {
38 To32Bit(output).device(d) = To32Bit(input).pad(paddings, pad_value);
40 output.device(d) = input.pad(paddings, pad_value);
|
| pad_op.cc | 73 T pad_value = T(); variable
80 pad_value = context->input(2).scalar<T>()();
129 pad_value, &collapsed_output);
138 OperateWithVariableRank(context, fixed_dims, in0, paddings, pad_value,
215 T pad_value, Tensor* output) {
219 Operate<0>(context, input.tensor<T, 0>(), paddings, pad_value, output);
224 Operate<1>(context, input.flat<T>(), paddings, pad_value, output);
227 Operate<2>(context, input.tensor<T, 2>(), paddings, pad_value, output);
230 Operate<3>(context, input.tensor<T, 3>(), paddings, pad_value, output);
233 Operate<4>(context, input.tensor<T, 4>(), paddings, pad_value, output)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| generator_io.py | 42 pad_value=None):
80 pad_value: default value for dynamic padding of data samples, if provided.
131 pad_value=pad_value)
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| shape_ops.py | 56 def frame(signal, frame_length, frame_step, pad_end=False, pad_value=0, axis=-1,
65 dimension are padded with `pad_value` until the window moves fully past the
83 pad_end: Whether to pad the end of `signal` with `pad_value`.
84 pad_value: An optional scalar `Tensor` to use where the input signal
94 ValueError: If `frame_length`, `frame_step`, `pad_value`, or `axis` are not
98 pad_value]):
123 # If padding is requested, pad the input signal tensor with pad_value.
125 pad_value = ops.convert_to_tensor(pad_value, signal.dtype)
126 pad_value.shape.assert_has_rank(0
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| shape_ops_test.py | 87 # Non-scalar pad_value.
89 shape_ops.frame([1], 1, 1, pad_end=True, pad_value=[1])
99 pad_end=True, pad_value=99).eval()
114 pad_end=True, pad_value=99)
127 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)
172 # With padding, we pad the last frame with pad_value.
174 pad_end=True, pad_value=99).eval()
199 # With padding, we pad the last frame with pad_value
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| pad_test.cc | 168 const float pad_value = 1.5f; local
171 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b),
175 expected->Fill(pad_value);
208 const float pad_value = -5.123f; local
214 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config);
217 expected_array.Fill(pad_value);
250 const float pad_value = -5.123f; local
252 input_array.Fill(pad_value);
260 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config);
263 expected_array.Fill(pad_value);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 3036 float pad_value; member in struct:xla::__anon44290::ConvPaddingTestcase
3101 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( local
3207 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( local
3586 HloInstruction* pad_value = builder.AddInstruction( local
3668 HloInstruction* pad_value = builder.AddInstruction( local
[all...] |
| algebraic_simplifier.cc | [all...] |
| hlo_evaluator_test.cc | 578 auto pad_value = LiteralUtil::CreateR0<int32>(kPadValue); local
580 b.AddInstruction(HloInstruction::CreateConstant(std::move(pad_value)));
604 auto pad_value = LiteralUtil::CreateR0<float>(kPadValue); local
606 b.AddInstruction(HloInstruction::CreateConstant(std::move(pad_value)));
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| legacy_optimized_ops.h | 1833 const T pad_value = static_cast<T>(0); local
[all...] |
| optimized_ops.h | 5913 const T pad_value = *pad_value_ptr; local
6070 const T pad_value = *pad_value_ptr; local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| types.h | 259 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value)
266 SetDim(i, pad_value);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| legacy_reference_ops.h | 2002 const T pad_value = static_cast<T>(0); local
[all...] |
| reference_ops.h | [all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| types.h | 274 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value)
282 SetDim(i, pad_value);
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 6238 const T pad_value = *pad_value_ptr; local
[all...] |
