| /external/tensorflow/tensorflow/core/kernels/ |
| D | mirror_pad_op.cc | 38 template <typename Device, typename T, typename Tpaddings>
85 typename TTypes<Tpaddings>::ConstMatrix paddings = in1.matrix<Tpaddings>(); in Compute()
87 const Tpaddings before = paddings(d, 0); // Pad before existing elements. in Compute()
88 const Tpaddings after = paddings(d, 1); // Pad after existing elements. in Compute()
124 functor::MirrorPad<Device, T, Tpaddings, i>()( \ in Compute()
155 #define DECLARE_CPU_SPEC(T, Tpaddings, i) \ argument
157 void MirrorPad<CpuDevice, T, Tpaddings, i>::operator()( \
160 TTypes<Tpaddings>::ConstMatrix, int); \
161 extern template struct MirrorPad<CpuDevice, T, Tpaddings, i>;
204 #define DECLARE_GPU_SPEC(T, Tpaddings, i) \ argument
[all …]
|
| D | mirror_pad_op.h | 339 template <typename Device, typename T, typename Tpaddings, int Dims>
344 typename TTypes<Tpaddings>::ConstMatrix padding, int offset) {
366 template <typename Device, typename T, typename Tpaddings, int Dims>
371 typename TTypes<Tpaddings>::ConstMatrix paddings, int offset,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | cpu_supported_ops.md | 146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
156 `Pad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
157 `PadV2` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
221 `SpaceToBatch` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
222 `SpaceToBatchND` | `Tblock_shape={int32,int64}`<br>`Tpaddings={int32,int64}`<b…
|
| D | gpu_supported_ops.md | 146 `MirrorPad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
156 `Pad` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
157 `PadV2` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
218 `SpaceToBatch` | `Tpaddings={int32,int64}`<br>`T={bool,complex64,double,floa…
219 `SpaceToBatchND` | `Tblock_shape={int32,int64}`<br>`Tpaddings={int32,int64}`<b…
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 2089 DataType Tpaddings; in CopyAttrsPadWithConv2D() local
2105 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "Tpaddings", &Tpaddings)); in CopyAttrsPadWithConv2D()
2118 nb->Attr("Tpaddings", Tpaddings); in CopyAttrsPadWithConv2D()
2124 DataType Tpaddings; in CopyAttrsPadWithFusedConv2D() local
2129 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "Tpaddings", &Tpaddings)); in CopyAttrsPadWithFusedConv2D()
2135 nb->Attr("Tpaddings", Tpaddings); in CopyAttrsPadWithFusedConv2D()
2144 DataType Tpaddings; in CopyAttrsFromPadAndConv2D() local
2161 TF_CHECK_OK(GetNodeAttr(orig_node2->def(), "Tpaddings", &Tpaddings)); in CopyAttrsFromPadAndConv2D()
2170 nb->Attr("Tpaddings", Tpaddings); in CopyAttrsFromPadAndConv2D()
2184 DataType Tpaddings; in CopyAttrsFromPadAndFusedConv2D() local
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 19832 type_attr: "Tpaddings"
19843 name: "Tpaddings"
19874 type_attr: "Tpaddings"
19885 name: "Tpaddings"
21743 type_attr: "Tpaddings"
21754 name: "Tpaddings"
21775 type_attr: "Tpaddings"
21790 name: "Tpaddings"
34952 type_attr: "Tpaddings"
34963 name: "Tpaddings"
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 18049 type_attr: "Tpaddings"
18060 name: "Tpaddings"
18120 type_attr: "Tpaddings"
18131 name: "Tpaddings"
19361 type_attr: "Tpaddings"
19372 name: "Tpaddings"
31735 type_attr: "Tpaddings"
31746 name: "Tpaddings"
31777 type_attr: "Tpaddings"
31801 name: "Tpaddings"
|
| D | ops_history.v2.pbtxt | 40097 type_attr: "Tpaddings"
40108 name: "Tpaddings"
40139 type_attr: "Tpaddings"
40150 name: "Tpaddings"
42615 type_attr: "Tpaddings"
42626 name: "Tpaddings"
42647 type_attr: "Tpaddings"
42662 name: "Tpaddings"
70618 type_attr: "Tpaddings"
70629 name: "Tpaddings"
[all …]
|
| D | ops_history.v1.pbtxt | 40535 type_attr: "Tpaddings"
40546 name: "Tpaddings"
40577 type_attr: "Tpaddings"
40588 name: "Tpaddings"
43138 type_attr: "Tpaddings"
43149 name: "Tpaddings"
43170 type_attr: "Tpaddings"
43185 name: "Tpaddings"
71245 type_attr: "Tpaddings"
71256 name: "Tpaddings"
[all …]
|