Home
last modified time | relevance | path

Searched refs:images_t (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dimage_ops.cc72 const Tensor& images_t = ctx->input(0); in Compute() local
74 OP_REQUIRES(ctx, images_t.shape().dims() == 4, in Compute()
78 (transform_t.dim_size(0) == images_t.dim_size(0) || in Compute()
103 out_height = images_t.shape().dim_size(1); in Compute()
104 out_width = images_t.shape().dim_size(2); in Compute()
110 TensorShape({images_t.dim_size(0), out_height, in Compute()
111 out_width, images_t.dim_size(3)}), in Compute()
114 auto images = images_t.tensor<T, 4>(); in Compute()
Dsegmentation_ops.cc44 const Tensor& images_t = ctx->input(0); in Compute() local
45 OP_REQUIRES(ctx, images_t.shape().dims() == 3, in Compute()
49 images_t.shape(), &forest_t)); in Compute()
51 images_t.shape(), &rank_t)); in Compute()
53 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, images_t.shape(), &output_t)); in Compute()
62 const auto images = images_t.tensor<T, 3>(); in Compute()
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py1096 def resize_fn(images_t, new_size): argument
1100 images_t, new_size, align_corners=align_corners)
1104 images_t, new_size, align_corners=align_corners)
1107 images_t, new_size, align_corners=align_corners)
1110 images_t, new_size, align_corners=align_corners)
1205 def resize_fn(images_t, new_size): argument
1215 math_ops.cast(array_ops.shape(images_t)[1:3], dtype=dtypes.float32))
1217 images_t,
1229 images_t, new_size, half_pixel_centers=True)
1232 images_t, new_size, half_pixel_centers=True)
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc1108 Tensor images_t = MakeRandomTensor({batch, height, width, in_depth}); in Conv2D() local
1111 Node* images = test::graph::Constant(graph, images_t, "images"); in Conv2D()
1236 Tensor images_t = MakeRandomTensor({batch, height, width, in_depth}); local
1240 Node* images = test::graph::Constant(graph, images_t, "images");
1268 Tensor images_t = MakeRandomTensor({batch, height, width, in_depth}); local
1275 Node* images = test::graph::Constant(graph, images_t, "images");