HomeSort by relevance Sort by last modified time
    Searched refs:grad_size (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_resize_ops.cc 357 absl::Span<const int64> grad_size, const int64 channels,
360 ComputeResizeConvolutionParameters(in_size, grad_size, align_corners);
385 builder, kernel, /*spatial_dimensions_offset=*/0, in_size, grad_size);
404 if (in_size[0] == 1 && grad_size[0] > 1) {
407 {grad_size[0]});
409 if (in_size[1] == 1 && grad_size[1] > 1) {
412 in_size, grad_size);
432 // If in_size[i] > 1 and grad_size[i] == 1, pad the output in dimension i.
437 if (in_size[i] > 1 && grad_size[i] == 1) {
624 const std::vector<int64> grad_size = {grad_shape.dim_size(1) variable
    [all...]
  /external/mesa3d/src/compiler/glsl/
builtin_functions.cpp 5152 int grad_size = coord_size - (sampler_type->sampler_array ? 1 : 0); local
    [all...]

Completed in 229 milliseconds