Searched refs:pad_op (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 82 const auto* pad_op = has_expand_op in ResolveDilatedConv() local
86 if (pad_op->type == OperatorType::kPad) { in ResolveDilatedConv()
88 CHECK_EQ(pad_op->inputs.size(), 2); in ResolveDilatedConv()
89 CHECK_EQ(pad_op->outputs.size(), 1); in ResolveDilatedConv()
95 ? GetOpWithInput(*model, pad_op->outputs[0]) in ResolveDilatedConv()
155 DeleteArrayIfUnused(pad_op->outputs[0], model); in ResolveDilatedConv()
156 std::vector<string> pad_op_inputs = pad_op->inputs; in ResolveDilatedConv()
157 model->operators.erase(FindOp(*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/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 | 235 pad_op = array_ops.pad(resize_op, [[0, 0], [1, 1], [2, 2], [0, 0]],
241 pad_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
295 pad_op = array_ops.pad(input_op, [[0, 0], [1, 1], [2, 2], [0, 0]],
301 pad_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 1075 pad_op = array_ops.pad(labeled_tensor.tensor,
1080 return core.LabeledTensor(pad_op, new_axes)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 62 "pad_op.cc",
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_test.cc | 790 TfLiteRegistration* pad_op = tflite::ops::builtin::Register_PADV2(); in TEST() local
792 interpreter.AddNodeWithParameters({0, 1}, {2}, nullptr, 0, nullptr, pad_op); in TEST()
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 169 tensorflow/core/kernels/pad_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 895 ":pad_op",
1039 name = "pad_op",
1040 prefix = "pad_op",
5616 "pad_op.h",
5779 "pad_op.cc",
|