Searched defs:grad_size (Results 1 – 3 of 3) sorted by relevance
380 absl::Span<const int64> grad_size, const int64 channels, in ResizeUsingDilationAndConvolutionGradOp()660 const std::vector<int64> grad_size = {grad_shape.dim_size(1), in Compile() local
513 const Tindex grad_size = grad.size(); in operator ()() local574 const Tindex grad_size = grad.size(); in operator ()() local784 const Tindex grad_size = grad.size(); in operator ()() local860 const Tindex grad_size = grad.size(); in operator ()() local
6509 int grad_size = coord_size - (sampler_type->sampler_array ? 1 : 0); in _texture() local