Home
last modified time | relevance | path

Searched refs:resize_op (Results 1 – 10 of 10) sorted by relevance

/external/XNNPACK/src/operators/
Dresize-bilinear-nchw.c30 xnn_operator_t resize_op = NULL; in xnn_create_resize_bilinear2d_nchw_f32() local
66 resize_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_resize_bilinear2d_nchw_f32()
67 if (resize_op == NULL) { in xnn_create_resize_bilinear2d_nchw_f32()
74 resize_op->channels = channels; in xnn_create_resize_bilinear2d_nchw_f32()
75 resize_op->input_pixel_stride = input_pixel_stride; in xnn_create_resize_bilinear2d_nchw_f32()
76 resize_op->output_pixel_stride = output_pixel_stride; in xnn_create_resize_bilinear2d_nchw_f32()
78 resize_op->type = xnn_operator_type_resize_bilinear_nchw_f32; in xnn_create_resize_bilinear2d_nchw_f32()
79 resize_op->flags = flags; in xnn_create_resize_bilinear2d_nchw_f32()
81 resize_op->state = xnn_run_state_invalid; in xnn_create_resize_bilinear2d_nchw_f32()
83 *resize_op_out = resize_op; in xnn_create_resize_bilinear2d_nchw_f32()
[all …]
Dresize-bilinear-nhwc.c30 xnn_operator_t resize_op = NULL; in xnn_create_resize_bilinear2d_nhwc_f32() local
66 resize_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_resize_bilinear2d_nhwc_f32()
67 if (resize_op == NULL) { in xnn_create_resize_bilinear2d_nhwc_f32()
74 resize_op->channels = channels; in xnn_create_resize_bilinear2d_nhwc_f32()
75 resize_op->input_pixel_stride = input_pixel_stride; in xnn_create_resize_bilinear2d_nhwc_f32()
76 resize_op->output_pixel_stride = output_pixel_stride; in xnn_create_resize_bilinear2d_nhwc_f32()
78 resize_op->type = xnn_operator_type_resize_bilinear_nhwc_f32; in xnn_create_resize_bilinear2d_nhwc_f32()
79 resize_op->flags = flags; in xnn_create_resize_bilinear2d_nhwc_f32()
81 resize_op->state = xnn_run_state_invalid; in xnn_create_resize_bilinear2d_nhwc_f32()
83 *resize_op_out = resize_op; in xnn_create_resize_bilinear2d_nhwc_f32()
[all …]
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py464 resize_op = node_from_map(input_node_map, mirror_pad_op.input[0])
465 if resize_op.op != "ResizeBilinear":
466 resize_op = None
470 resize_op = input_op
472 resize_op = None
475 if not mirror_pad_op and not resize_op:
482 if resize_op:
483 node_reference_count[resize_op.name] -= 1
486 if resize_op:
508 if resize_op:
[all …]
Doptimize_for_inference_test.py302 resize_op = image_ops.resize_bilinear(
304 pad_op = array_ops.pad(resize_op, [[0, 0], [1, 1], [2, 2], [0, 0]],
334 resize_op = image_ops.resize_bilinear(
340 resize_op, weights_op, [1, 1, 1, 1], padding="VALID", name="output")
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions_test.cc55 Output resize_op = ResizeBilinear(root.WithOpName("resize_op"), input_op, in TestFuseResizePadAndConv() local
64 MirrorPad(root.WithOpName("pad_op"), resize_op, pad_dims_op, "REFLECT"); in TestFuseResizePadAndConv()
112 Output resize_op = ResizeBilinear(root.WithOpName("resize_op"), input_op, in TestFuseResizeAndConv() local
122 Output conv_op = Conv2D(root.WithOpName("output"), resize_op, weights_op, in TestFuseResizeAndConv()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h58 const TfLiteRegistration* resize_op = in resize() local
64 interpreter->AddNodeWithParameters({0, 1}, {2}, nullptr, 0, params, resize_op, in resize()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1416 tensorflow::NodeDef* resize_op = tensorflow_graph->add_node(); in ConvertResizeBilinearOperator() local
1417 resize_op->set_op("ResizeBilinear"); in ConvertResizeBilinearOperator()
1418 resize_op->set_name(src_op.outputs[0]); in ConvertResizeBilinearOperator()
1420 *resize_op->add_input() = src_op.inputs[0]; in ConvertResizeBilinearOperator()
1421 *resize_op->add_input() = src_op.inputs[1]; in ConvertResizeBilinearOperator()
1422 (*resize_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertResizeBilinearOperator()
1423 (*resize_op->mutable_attr())["align_corners"].set_b(src_op.align_corners); in ConvertResizeBilinearOperator()
1424 (*resize_op->mutable_attr())["half_pixel_centers"].set_b( in ConvertResizeBilinearOperator()
1431 tensorflow::NodeDef* resize_op = tensorflow_graph->add_node(); in ConvertResizeNearestNeighborOperator() local
1432 resize_op->set_op("ResizeNearestNeighbor"); in ConvertResizeNearestNeighborOperator()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc642 void CreateCondTrueBranch(TF::TensorListResizeOp resize_op, Type shape_dtype, in CreateCondTrueBranch()
654 Location loc = resize_op.getLoc(); in CreateCondTrueBranch()
663 loc, resize_op.output_handle().getType(), elem_shape, size_diff); in CreateCondTrueBranch()
/external/XNNPACK/include/
Dxnnpack.h1515 xnn_operator_t resize_op,
1533 xnn_operator_t resize_op,
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_test.py762 resize_op = image_ops.resize_bilinear(
764 deps = [resize_op]
810 resize_op = image_ops.resize_bicubic(
812 deps = [resize_op]
868 resize_op = image_ops.resize_area(img, [299, 299], align_corners=False)
869 deps = [resize_op]