Home
last modified time | relevance | path

Searched refs:in_y (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/image/
Dresize_bilinear_op_gpu.cu.cc52 const float in_y = (static_cast<float>(y) + 0.5f) * height_scale - 0.5f; in ResizeBilinearKernel_faster() local
54 const int top_y_index = in_y > 0.0 ? floorf(in_y) : 0; in ResizeBilinearKernel_faster()
56 (in_y < in_height - 1) ? ceilf(in_y) : in_height - 1; in ResizeBilinearKernel_faster()
57 const float y_lerp = in_y - floorf(in_y); in ResizeBilinearKernel_faster()
131 const float in_y = (static_cast<float>(y) + 0.5f) * height_scale - 0.5f; in ResizeBilinearKernel() local
133 const int top_y_index = in_y > 0.0 ? floorf(in_y) : 0; in ResizeBilinearKernel()
135 (in_y < in_height - 1) ? ceilf(in_y) : in_height - 1; in ResizeBilinearKernel()
136 const float y_lerp = in_y - floorf(in_y); in ResizeBilinearKernel()
260 int in_y = in_y_start; in ResizeBilinearDeterministicGradKernel() local
261 while (out_y < out_y_center + 1 && in_y < resized_height) { in ResizeBilinearDeterministicGradKernel()
[all …]
Dcrop_and_resize_op_gpu.cu.cc72 const float in_y = (crop_height > 1) in CropAndResizeKernel() local
75 if (in_y < 0 || in_y > image_height - 1) { in CropAndResizeKernel()
89 const int top_y_index = floorf(in_y); in CropAndResizeKernel()
90 const int bottom_y_index = ceilf(in_y); in CropAndResizeKernel()
91 const float y_lerp = in_y - top_y_index; in CropAndResizeKernel()
122 const int closest_y_index = roundf(in_y); in CropAndResizeKernel()
165 const float in_y = (crop_height > 1) in CropAndResizeBackpropImageKernel() local
168 if (in_y < 0 || in_y > image_height - 1) { in CropAndResizeBackpropImageKernel()
180 const int top_y_index = floorf(in_y); in CropAndResizeBackpropImageKernel()
181 const int bottom_y_index = ceilf(in_y); in CropAndResizeBackpropImageKernel()
[all …]
Dresize_nearest_neighbor_op_gpu.cu.cc50 const int in_y = in ResizeNearestNeighborNHWC() local
60 const int idx = (in_y * in_width + in_x) * channels + c; in ResizeNearestNeighborNHWC()
81 const int in_y = in LegacyResizeNearestNeighborNHWC() local
89 const int idx = (in_y * in_width + in_x) * channels + c; in LegacyResizeNearestNeighborNHWC()
106 int in_y = n % in_height; in ResizeNearestNeighborBackwardNHWC() local
112 floorf((static_cast<float>(in_y) + 0.5f) * height_scale)), in ResizeNearestNeighborBackwardNHWC()
137 int in_y = n % in_height; in LegacyResizeNearestNeighborBackwardNHWC() local
142 min((align_corners) ? static_cast<int>(roundf(in_y * height_scale)) in LegacyResizeNearestNeighborBackwardNHWC()
143 : static_cast<int>(floorf(in_y * height_scale)), in LegacyResizeNearestNeighborBackwardNHWC()
Dcrop_and_resize_op.cc258 const float in_y = (crop_height > 1) in operator ()() local
261 if (in_y < 0 || in_y > image_height - 1) { in operator ()()
270 const int top_y_index = floorf(in_y); in operator ()()
271 const int bottom_y_index = ceilf(in_y); in operator ()()
272 const float y_lerp = in_y - top_y_index; in operator ()()
315 const int closest_y_index = roundf(in_y); in operator ()()
484 const float in_y = (crop_height > 1) in operator ()() local
487 if (in_y < 0 || in_y > image_height - 1) { in operator ()()
490 const int top_y_index = floorf(in_y); in operator ()()
491 const int bottom_y_index = ceilf(in_y); in operator ()()
[all …]
Dresize_area_op_test.cc105 const float in_y = y * height_scale; in ResizeAreaBaseline() local
109 int64 y_start = std::floor(in_y); in ResizeAreaBaseline()
122 float scale_y = i < in_y in ResizeAreaBaseline()
123 ? (i + 1 > in_y1 ? height_scale : i + 1 - in_y) in ResizeAreaBaseline()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dtranspose_conv.h68 for (int in_y = 0; in_y < input_height; ++in_y) { in TransposeConv() local
73 const int out_y_origin = (in_y * stride_height) - pad_height; in TransposeConv()
85 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
159 for (int in_y = 0; in_y < input_height; ++in_y) { in TransposeConv() local
164 const int out_y_origin = (in_y * stride_height) - pad_height; in TransposeConv()
176 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
Dconv.h65 const int in_y = in_y_origin + dilation_height_factor * filter_y; in Conv() local
71 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in Conv()
72 (in_y < input_height); in Conv()
79 float input_value = input_data[Offset(input_shape, batch, in_y, in Conv()
148 const int in_y = in_y_origin + dilation_height_factor * filter_y; in Conv() local
154 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in Conv()
155 (in_y < input_height); in Conv()
162 int32_t input_val = input_data[Offset(input_shape, batch, in_y, in Conv()
231 const int in_y = in HybridConvPerChannel() local
235 if ((in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in HybridConvPerChannel()
[all …]
Dpooling.h63 const int in_y = in_y_origin + filter_y; in AveragePool() local
65 input_data[Offset(input_shape, batch, in_y, in_x, channel)]; in AveragePool()
119 const int in_y = in_y_origin + filter_y; in AveragePool() local
121 input_data[Offset(input_shape, batch, in_y, in_x, channel)]; in AveragePool()
172 const int in_y = in_y_origin + filter_y; in L2Pool() local
174 input_data[Offset(input_shape, batch, in_y, in_x, channel)]; in L2Pool()
225 const int in_y = in_y_origin + filter_y; in MaxPool() local
228 input_data[Offset(input_shape, batch, in_y, in_x, channel)]); in MaxPool()
279 const int in_y = in_y_origin + filter_y; in MaxPool() local
282 input_data[Offset(input_shape, batch, in_y, in_x, channel)]); in MaxPool()
Ddepthwiseconv_uint8.h142 const int in_y = in Run() local
146 if ((in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in Run()
147 (in_y < input_height)) { in Run()
149 input_data[Offset(input_shape, b, in_y, in_x, ic)]; in Run()
229 const int in_y = in RunPerChannel() local
233 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in RunPerChannel()
234 (in_y < input_height); in RunPerChannel()
237 input_shape, batch, in_y, in_x, in_channel)]; in RunPerChannel()
Ddepthwiseconv_float.h68 const int in_y = in DepthwiseConv() local
72 if ((in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConv()
73 (in_y < input_height)) { in DepthwiseConv()
75 input_data[Offset(input_shape, b, in_y, in_x, ic)]; in DepthwiseConv()
Dconv3d.h67 const int in_y = in Conv3D() local
75 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in Conv3D()
76 (in_y < input_height) && (in_d >= 0) && in Conv3D()
86 input_shape, batch, in_d, in_y, in_x, in_channel)]; in Conv3D()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h73 const int in_y = in DepthwiseConvPerChannel() local
77 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConvPerChannel()
78 (in_y < input_height); in DepthwiseConvPerChannel()
81 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
170 const int in_y = in DepthwiseConvPerChannel() local
174 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConvPerChannel()
175 (in_y < input_height); in DepthwiseConvPerChannel()
178 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
254 const int in_y = in DepthwiseConvHybridPerChannel() local
258 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConvHybridPerChannel()
[all …]
Dtranspose_conv.h67 for (int in_y = 0; in_y < input_height; ++in_y) { in TransposeConv() local
72 const int out_y_origin = (in_y * stride_height) - pad_height; in TransposeConv()
84 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
164 for (int in_y = 0; in_y < input_height; ++in_y) { in TransposeConv() local
169 const int out_y_origin = (in_y * stride_height) - pad_height; in TransposeConv()
181 input_shape, batch, in_y, in_x, in_channel)]; in TransposeConv()
Dconv.h72 const int in_y = in_y_origin + dilation_height_factor * filter_y; in ConvPerChannel() local
78 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in ConvPerChannel()
79 (in_y < input_height); in ConvPerChannel()
86 int32_t input_val = input_data[Offset(input_shape, batch, in_y, in ConvPerChannel()
175 const int in_y = in_y_origin + dilation_height_factor * filter_y; in ConvPerChannel() local
181 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in ConvPerChannel()
182 (in_y < input_height); in ConvPerChannel()
189 int32_t input_val = input_data[Offset(input_shape, batch, in_y, in ConvPerChannel()
Dpooling.h63 const int in_y = in_y_origin + filter_y; in AveragePool() local
65 input_data[Offset(input_shape, batch, in_y, in_x, channel)]; in AveragePool()
123 const int in_y = in_y_origin + filter_y; in MaxPool() local
126 input_data[Offset(input_shape, batch, in_y, in_x, channel)]); in MaxPool()
179 const int in_y = in_y_origin + filter_y; in AveragePool() local
181 input_data[Offset(input_shape, batch, in_y, in_x, channel)]; in AveragePool()
239 const int in_y = in_y_origin + filter_y; in MaxPool() local
242 input_data[Offset(input_shape, batch, in_y, in_x, channel)]); in MaxPool()
/external/tensorflow/tensorflow/lite/kernels/perception/
Ddense_image_warp.cc49 for (int in_y = 0; in_y < height; ++in_y) { in DenseImageWarp() local
52 in_y - flow_data[Offset(flow_shape, batch, in_y, in_x, 0)]; in DenseImageWarp()
54 in_x - flow_data[Offset(flow_shape, batch, in_y, in_x, 1)]; in DenseImageWarp()
81 output_data[Offset(input_shape, batch, in_y, in_x, c)] = interp; in DenseImageWarp()
Dmax_unpooling_2d.cc42 for (int in_y = 0; in_y < input_shape.Dims(1); ++in_y) { in MaxUnpooling() local
46 Offset(input_shape, batch, in_y, in_x, channel); in MaxUnpooling()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_impl.h78 const Tensor in_y, bool adj_x, bool adj_y, bool trans_x, in Run()
83 auto Ty = in_y.tensor<Scalar, 3>(); in Run()
123 const Tensor& in_y, bool adj_x, bool adj_y, bool trans_x,
132 auto Ty = in_y.flat_inner_dims<Scalar, 2>();
137 auto Ty = in_y.tensor<Scalar, 3>();
178 static void Run(const Tensor& in_x, const Tensor& in_y, bool adj_x,
188 auto y = ConstTensorSliceToEigenMatrix(in_y, y_batch_index);
229 const Tensor& in_y, bool adj_x, bool adj_y, bool trans_x,
252 ParallelMatMulKernel::Run(context, in_x, in_y, adj_x, adj_y, trans_x,
260 [&in_x, &in_y, adj_x, adj_y, trans_x, trans_y, &bcast, out](
[all …]
Dconv_ops_using_gemm.cc162 const int in_y = in_y_origin + filter_y; in operator ()() local
166 if ((in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in operator ()()
167 (in_y < input_height)) { in operator ()()
171 (in_y * input_width * input_depth) + in operator ()()
347 const int in_y = in_y_origin + filter_y; in operator ()() local
352 if ((in_y < 0) || (in_y >= input_height)) { in operator ()()
387 input_batch_start + (in_y * input_width * input_depth) + in operator ()()
Dconv_ops_fused_image_transform.cc188 float in_y = (cache_y - top_padding); in CalculatePerCacheLineParameters() local
189 if (in_y < 0) { in CalculatePerCacheLineParameters()
190 in_y = -(in_y + 1.0f - pad_offset); in CalculatePerCacheLineParameters()
191 } else if (in_y >= resized_height) { in CalculatePerCacheLineParameters()
192 in_y = (resized_height * 2.0f) - (in_y + 1.0f + pad_offset); in CalculatePerCacheLineParameters()
195 in_y *= st.height_scale; in CalculatePerCacheLineParameters()
196 const int64 top_y_index = static_cast<int64>(std::floor(in_y)); in CalculatePerCacheLineParameters()
198 std::min(static_cast<int64>(std::ceil(in_y)), (st.in_height - 1)); in CalculatePerCacheLineParameters()
200 result.y_lerp = static_cast<T1>(in_y - top_y_index); in CalculatePerCacheLineParameters()
Dquantized_conv_ops.cc137 const int in_y = in_y_origin + filter_y; in operator ()() local
141 if ((in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in operator ()()
142 (in_y < input_height)) { in operator ()()
146 (in_y * input_width * input_depth) + in operator ()()
313 const int in_y = in_y_origin + filter_y; in operator ()() local
318 if ((in_y < 0) || (in_y >= input_height)) { in operator ()()
355 input_batch_start + (in_y * input_width * input_depth) + in operator ()()
/external/deqp/external/openglcts/modules/glesext/tessellation_shader/
DesextcTessellationShaderUtils.hpp52 _ivec4(int in_x, int in_y, int in_z, int in_w) in _ivec4()
55 y = in_y; in _ivec4()
92 _vec2(float in_x, float in_y) in _vec2()
95 y = in_y; in _vec2()
134 _vec4(float in_x, float in_y, float in_z, float in_w) in _vec4()
137 y = in_y; in _vec4()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_triangular_solve_op_impl.h84 static void Run(const Tensor& in_x, const Tensor& in_y, bool lower, in Run()
94 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()
121 const Tensor& in_y, bool adjoint, bool lower,
126 in_x.dim_size(1) * in_x.dim_size(1) * in_y.dim_size(2) / 2;
142 [&in_x, &in_y, adjoint, lower, &bcast, out](int start, int limit) {
144 in_x, in_y, lower, adjoint, bcast, out, start, limit);
258 const Tensor& in_y, bool adjoint, bool lower,
266 if (!bcast.IsBroadcastingRequired() || out->shape() == in_y.shape()) {
267 auto src_device_mem = AsDeviceMemory(in_y.template flat<Scalar>().data());
280 auto* b_base_ptr = in_y.template flat<Scalar>().data();
Dbanded_triangular_solve_op.cc62 static void Run(const Tensor& in_x, const Tensor& in_y, bool lower, in Run()
75 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()
169 const Tensor& in_y, bool adjoint, bool lower, in Launch()
174 in_x.dim_size(1) * in_x.dim_size(2) * in_y.dim_size(2); in Launch()
196 [&in_x, &in_y, adjoint, lower, &bcast, out](int64 start, in Launch()
199 in_x, in_y, lower, adjoint, bcast, out, start, limit); in Launch()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc92 const int in_y = in_y_origin + dilation_height_factor * filter_y; in DepthwiseConvPerChannel() local
97 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConvPerChannel()
98 (in_y < input_height); in DepthwiseConvPerChannel()
109 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConvPerChannel()
215 const int in_y = in_y_origin + dilation_height_factor * filter_y; in DepthwiseConv() local
220 (in_x >= 0) && (in_x < input_width) && (in_y >= 0) && in DepthwiseConv()
221 (in_y < input_height); in DepthwiseConv()
232 input_shape, batch, in_y, in_x, in_channel)]; in DepthwiseConv()

123