| /external/tensorflow/tensorflow/core/kernels/image/ |
| 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 | 345 template <typename Device, typename T, typename Tpaddings, int Dims>
350 typename TTypes<Tpaddings>::ConstMatrix padding, int offset) {
372 template <typename Device, typename T, typename Tpaddings, int Dims>
377 typename TTypes<Tpaddings>::ConstMatrix paddings, int offset,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Pad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | PadV2.pbtxt | 9 type_attr: "Tpaddings"
24 name: "Tpaddings"
|
| D | SpaceToBatch.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | MirrorPad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | MirrorPadGrad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | RiscPad.pbtxt | 9 type_attr: "Tpaddings"
32 name: "Tpaddings"
|
| D | SpaceToBatchND.pbtxt | 13 type_attr: "Tpaddings"
37 name: "Tpaddings"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Pad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | PadV2.pbtxt | 9 type_attr: "Tpaddings"
24 name: "Tpaddings"
|
| D | SpaceToBatch.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | MirrorPad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | MirrorPadGrad.pbtxt | 9 type_attr: "Tpaddings"
20 name: "Tpaddings"
|
| D | SpaceToBatchND.pbtxt | 13 type_attr: "Tpaddings"
37 name: "Tpaddings"
|
| /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/common_runtime/ |
| D | mkl_layout_pass.cc | 2671 DataType Tpaddings; in CopyAttrsFromPadAndConv2D() local
2688 TF_CHECK_OK(GetNodeAttr(orig_node2->def(), "Tpaddings", &Tpaddings)); in CopyAttrsFromPadAndConv2D()
2697 nb->Attr("Tpaddings", Tpaddings); in CopyAttrsFromPadAndConv2D()
2711 DataType Tpaddings; in CopyAttrsFromPadAndFusedConv2D() local
2725 TF_CHECK_OK(GetNodeAttr(pad->def(), "Tpaddings", &Tpaddings)); in CopyAttrsFromPadAndFusedConv2D()
2735 nb->Attr("Tpaddings", Tpaddings); in CopyAttrsFromPadAndFusedConv2D()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25315 type_attr: "Tpaddings"
25326 name: "Tpaddings"
25357 type_attr: "Tpaddings"
25368 name: "Tpaddings"
27493 type_attr: "Tpaddings"
27504 name: "Tpaddings"
27525 type_attr: "Tpaddings"
27540 name: "Tpaddings"
42614 type_attr: "Tpaddings"
42637 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 | 39434 type_attr: "Tpaddings"
39445 name: "Tpaddings"
39476 type_attr: "Tpaddings"
39487 name: "Tpaddings"
41952 type_attr: "Tpaddings"
41963 name: "Tpaddings"
41984 type_attr: "Tpaddings"
41999 name: "Tpaddings"
69955 type_attr: "Tpaddings"
69966 name: "Tpaddings"
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8436 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
8478 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
9356 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
9407 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
13959 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<1>;
14095 TF_DerivedOperandTypeAttr Tpaddings = TF_DerivedOperandTypeAttr<2>;
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | keras_imagenet_main.pbtxt | 26282 key: "Tpaddings"
34894 key: "Tpaddings"
|
| D | keras_imagenet_main_graph_mode.pbtxt | 26542 key: "Tpaddings"
|