Home
last modified time | relevance | path

Searched refs:Tpadding (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpad_op_gpu.cu.cc30 #define DEFINE_GPU_PAD_SPECS(T, Tpadding) \ argument
31 template struct functor::Pad<GPUDevice, T, Tpadding, 0>; \
32 template struct functor::Pad<GPUDevice, T, Tpadding, 1>; \
33 template struct functor::Pad<GPUDevice, T, Tpadding, 2>; \
34 template struct functor::Pad<GPUDevice, T, Tpadding, 3>; \
35 template struct functor::Pad<GPUDevice, T, Tpadding, 4>; \
36 template struct functor::Pad<GPUDevice, T, Tpadding, 5>; \
37 template struct functor::Pad<GPUDevice, T, Tpadding, 6>; \
38 template struct functor::Pad<GPUDevice, T, Tpadding, 7>; \
39 template struct functor::Pad<GPUDevice, T, Tpadding, 8>;
Dpad_op.h28 template <typename Device, typename T, typename Tpadding, int Dims>
34 Eigen::array<Eigen::IndexPair<Tpadding>, Dims> paddings, in operator()
45 template <typename Device, typename T, typename Tpadding>
46 struct Pad<Device, T, Tpadding, 0> {
50 Eigen::array<Eigen::IndexPair<Tpadding>, 0>, T) {
Dpad_op.cc42 template <typename Device, typename T, typename Tpadding>
79 typename TTypes<Tpadding>::ConstMatrix paddings = in1.matrix<Tpadding>(); in Compute()
81 const Tpadding before_d = in Compute()
83 const Tpadding after_d = paddings(d, 1); // Pad after existing elements. in Compute()
116 typename TTypes<Tpadding>::ConstMatrix collapsed_paddings_matrix = in Compute()
117 collapsed_paddings_ref.matrix<Tpadding>(); in Compute()
156 typename TTypes<Tpadding>::ConstMatrix paddings = in CollapseAdjacentNonPaddedDimensions()
157 paddings_as_tensor.matrix<Tpadding>(); in CollapseAdjacentNonPaddedDimensions()
195 collapsed_paddings_as_tensor->matrix<Tpadding>(); in CollapseAdjacentNonPaddedDimensions()
205 typename TTypes<Tpadding>::ConstMatrix paddings, in OperateWithVariableRank()
[all …]
/external/tensorflow/tensorflow/core/kernels/risc/experimental/
Drisc_pad_op.cc24 template <typename T, typename Tpadding>
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.cc414 typename Toutput, typename Ttemp_output, typename Tpadding,
753 Tpadding* paddings = nullptr; in PadWithConvFusion()
762 paddings = static_cast<Tpadding*>( in PadWithConvFusion()
763 const_cast<Tpadding*>(paddings_tf.flat<Tpadding>().data())); in PadWithConvFusion()
940 std::vector<Tpadding> padding_list_;
1157 typename Toutput, typename Ttemp_output, typename Tpadding,
1161 Tpadding, false, false, false, native_format> {
1165 Tpadding, false, false, false, native_format>(context) { in MklFusedConvOp()
1284 typename Toutput, typename Ttemp_output, typename Tpadding,
1289 Tpadding, bias_enabled, false, is_depthwise,
[all …]