Home
last modified time | relevance | path

Searched refs:output_tile_bounds (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h511 std::array<llvm::Value*, 3> output_tile_bounds; member
Dir_emitter_unnested.cc4550 tiling_kernel_info.output_tile_bounds[kDimX]), in EmitEpilogueForReduction()
4552 tiling_kernel_info.output_tile_bounds[kDimY])); in EmitEpilogueForReduction()
4730 std::array<llvm::Value*, 3> output_tile_bounds; in EmitTilingKernel() local
4738 output_tile_bounds[i] = in EmitTilingKernel()
4758 output_tile_bounds[1], output_tile_bounds[2], &ksl); in EmitTilingKernel()
4785 return {output_tile_bounds, tile_origin}; in EmitTilingKernel()