Searched refs:grad_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 357 absl::Span<const int64> grad_size, const int64 channels, in ResizeUsingDilationAndConvolutionGradOp() argument
360 ComputeResizeConvolutionParameters(in_size, grad_size, align_corners); in ResizeUsingDilationAndConvolutionGradOp()
385 builder, kernel, /*spatial_dimensions_offset=*/0, in_size, grad_size); in ResizeUsingDilationAndConvolutionGradOp()
404 if (in_size[0] == 1 && grad_size[0] > 1) { in ResizeUsingDilationAndConvolutionGradOp()
407 {grad_size[0]}); in ResizeUsingDilationAndConvolutionGradOp()
409 if (in_size[1] == 1 && grad_size[1] > 1) { in ResizeUsingDilationAndConvolutionGradOp()
412 in_size, grad_size); in ResizeUsingDilationAndConvolutionGradOp()
437 if (in_size[i] > 1 && grad_size[i] == 1) { in ResizeUsingDilationAndConvolutionGradOp()
624 const std::vector<int64> grad_size = {grad_shape.dim_size(1), in Compile() local
631 OP_REQUIRES(ctx, grad_size[0] > 0 && grad_size[1] > 0, in Compile()
[all …]
|
| /external/mesa3d/src/compiler/glsl/ |
| D | builtin_functions.cpp | 5152 int grad_size = coord_size - (sampler_type->sampler_array ? 1 : 0); in _texture() local
5153 ir_variable *dPdx = in_var(glsl_type::vec(grad_size), "dPdx"); in _texture()
5154 ir_variable *dPdy = in_var(glsl_type::vec(grad_size), "dPdy"); in _texture()
|