| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 85 const auto* pad_op = has_expand_op in ResolveDilatedConv() local
89 if (pad_op && pad_op->type == OperatorType::kPad) { in ResolveDilatedConv()
91 CHECK_EQ(pad_op->inputs.size(), 2); in ResolveDilatedConv()
92 CHECK_EQ(pad_op->outputs.size(), 1); in ResolveDilatedConv()
98 ? GetOpWithInput(*model, pad_op->outputs[0]) in ResolveDilatedConv()
134 pad_op = pre_stb_pad_op; in ResolveDilatedConv()
163 DeleteOpAndArrays(model, pad_op); in ResolveDilatedConv()
|
| D | resolve_pad_attributes.cc | 32 auto* pad_op = pad_it->get(); in Run() local
33 if (pad_op->type != OperatorType::kPad) return ::tensorflow::Status::OK(); in Run()
35 auto* op = static_cast<PadOperator*>(pad_op); in Run()
|
| D | resolve_padv2_attributes.cc | 32 auto* pad_op = pad_it->get(); in Run() local
33 if (pad_op->type != OperatorType::kPadV2) return ::tensorflow::Status::OK(); in Run()
35 auto* op = static_cast<PadV2Operator*>(pad_op); in Run()
|
| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 176 xnn_operator_t pad_op = nullptr; in TestX32() local
180 &padding_value, 0, &pad_op)); in TestX32()
181 ASSERT_NE(nullptr, pad_op); in TestX32()
184 …std::unique_ptr<xnn_operator, decltype(&xnn_delete_operator)> auto_pad_op(pad_op, xnn_delete_opera… in TestX32()
188 pad_op, in TestX32()
195 xnn_run_operator(pad_op, nullptr /* thread pool */)); in TestX32()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_space_to_depth_pass.cc | 284 auto pad_op = in HandleConv2DFilter() local
287 before_reshape_value = pad_op; in HandleConv2DFilter()
549 auto pad_op = dyn_cast_or_null<TF::PadOp>(next_input.getDefiningOp()); in GetInputBlockArgNum() local
552 while (pad_op || cast_op) { in GetInputBlockArgNum()
553 if (pad_op) { in GetInputBlockArgNum()
554 auto block_arg_num = GetBlockArgNum(pad_op.input()); in GetInputBlockArgNum()
556 next_input = pad_op.input(); in GetInputBlockArgNum()
562 pad_op = dyn_cast_or_null<TF::PadOp>(next_input.getDefiningOp()); in GetInputBlockArgNum()
600 if (auto pad_op = dyn_cast_or_null<TF::PadOp>(conv2d_input.getDefiningOp())) { in HandleFirstConvolution() local
602 if (failed(HandlePad(pad_op, filter_shape[0], block_size))) return; in HandleFirstConvolution()
[all …]
|
| D | legalize_hlo.cc | 1171 auto pad_op = cast<mhlo::PadOp>(old_op); in ConvertPadOp() local
1172 mlir::Location loc = pad_op.getLoc(); in ConvertPadOp()
1175 for (auto p : llvm::zip(pad_op.edge_padding_low().getValues<APInt>(), in ConvertPadOp()
1176 pad_op.edge_padding_high().getValues<APInt>())) { in ConvertPadOp()
1180 auto attr_type = RankedTensorType::get({pad_op.edge_padding_low().size(), 2}, in ConvertPadOp()
1184 return rewriter.create<PadV2Op>(loc, pad_op.getType(), pad_op.operand(), in ConvertPadOp()
1185 padding_op, pad_op.padding_value()); in ConvertPadOp()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_convolutions_test.cc | 63 Output pad_op = in TestFuseResizePadAndConv() local
72 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFuseResizePadAndConv()
165 Output pad_op = in TestFusePadAndConv() local
174 Output conv_op = Conv2D(root.WithOpName("output"), pad_op, weights_op, in TestFusePadAndConv()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_test.py | 304 pad_op = array_ops.pad(resize_op, [[0, 0], [1, 1], [2, 2], [0, 0]],
310 pad_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
364 pad_op = array_ops.pad(input_op, [[0, 0], [1, 1], [2, 2], [0, 0]],
370 pad_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 166 TF::PadOp pad_op; in matchAndRewrite() local
171 pad_op = llvm::cast<TF::PadOp>(next_op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 429 "//tensorflow/core/kernels:pad_op",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 74 "pad_op.cc",
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 1019 TfLiteRegistration* pad_op = tflite::ops::builtin::Register_PADV2(); in TEST() local
1021 interpreter.AddNodeWithParameters({0, 1}, {2}, nullptr, 0, nullptr, pad_op); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1652 auto pad_op = rewriter.create<tosa::PadOp>( in matchAndRewrite() local
1655 rewriter.replaceOp(op, {pad_op.getResult()}); in matchAndRewrite()
|
| D | legalize_tfl.cc | 2264 auto pad_op = rewriter.create<tosa::PadOp>( in matchAndRewrite() local
2267 rewriter.replaceOp(op, {pad_op.getResult()}); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 977 ":pad_op",
1119 name = "pad_op",
1120 prefix = "pad_op",
5927 "pad_op.h",
6160 "pad_op.cc",
|