Home
last modified time | relevance | path

Searched defs:pad_value (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpad.cc149 #define TF_LITE_PAD(type, op_name, scalar, pad_value) \ in Eval() argument
167 float pad_value = op_context.constant_values == nullptr in Eval() local
185 uint8_t pad_value; in Eval() local
218 int8_t pad_value; in Eval() local
243 int32_t pad_value = in Eval() local
254 int64_t pad_value = in Eval() local
Dspace_to_batch_nd.cc116 #define TF_LITE_SPACE_TO_BATCH_ND(type, scalar, pad_value) \ in Eval() argument
/external/tensorflow/tensorflow/core/kernels/
Dpad_op.cc73 T pad_value = T(); in Compute() local
215 T pad_value, Tensor* output) { in OperateWithVariableRank()
251 typename TTypes<Tpadding>::ConstMatrix paddings, T pad_value, in Operate()
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgenerator_io.py42 pad_value=None): argument
/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
/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc168 const float pad_value = 1.5f; in TEST_P() local
208 const float pad_value = -5.123f; in TEST_P() local
250 const float pad_value = -5.123f; in XLA_TEST_P() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h1925 const int32_t pad_value) { in SpaceToBatchND()
1968 const Dims<4>& output_dims, const T pad_value) { in PadV2()
1990 const Dims<4>& output_dims, const int32_t pad_value) { in Pad()
2002 const T pad_value = static_cast<T>(0); in Pad() local
Dreference_ops.h3104 const int32_t pad_value = params.output_offset; in SpaceToBatchND() local
3231 const T pad_value = *pad_value_ptr; in PadImpl() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3036 float pad_value; member
3101 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
3207 auto* pad_value = builder.AddInstruction(HloInstruction::CreateConstant( in TEST_P() local
3586 HloInstruction* pad_value = builder.AddInstruction( in TEST_F() local
3668 HloInstruction* pad_value = builder.AddInstruction( in TEST_F() local
Dhlo_evaluator_test.cc578 auto pad_value = LiteralUtil::CreateR0<int32>(kPadValue); in TEST_F() local
604 auto pad_value = LiteralUtil::CreateR0<float>(kPadValue); in TEST_P() local
Dalgebraic_simplifier.cc3188 const HloInstruction* pad_value = pad->operand(1); in HandleReduceWindow() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1800 const Dims<4>& output_dims, const T pad_value) { in PadV2()
1821 const Dims<4>& output_dims, const int32_t pad_value) { in Pad()
1833 const T pad_value = static_cast<T>(0); in Pad() local
Doptimized_ops.h5913 const T pad_value = *pad_value_ptr; in PadImpl() local
6070 const T pad_value = *pad_value_ptr; in PadImageStyleMemset() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h259 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h274 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h6238 const T pad_value = *pad_value_ptr; in PadImpl() local