Searched refs:output_tile_bounds (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 58 absl::Span<llvm::Value* const> output_tile_bounds,
|
| D | ir_emitter_unnested.cc | 3000 std::vector<llvm::Value*> output_tile_bounds(3); in EmitBlock() local
3009 output_tile_bounds[i] = in EmitBlock()
3016 emit_one_tile(tile_origin, output_tile_bounds, block_contains_multi_tiles); in EmitBlock()
3137 absl::Span<llvm::Value* const> output_tile_bounds, in EmitKernel()
3152 input_tile_origin, "input", output_tile_bounds[2], in EmitKernel()
3153 output_tile_bounds[1], in EmitKernel()
3178 output_tile_origin, "output", output_tile_bounds[1], in EmitKernel()
3179 output_tile_bounds[2], in EmitKernel()
|