Searched refs:req_block_sizes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_tiling.cc | 120 absl::Span<const int64> req_block_sizes, int64 num_threads_y, in KernelMappingScheme() argument
129 DCHECK_EQ(req_block_sizes.size(), 3);
135 block_sizes_.reserve(req_block_sizes.size());
136 absl::c_transform(req_block_sizes, dims_in_tiles_,
|
| D | kernel_tiling.h | 100 absl::Span<const int64> req_block_sizes,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3721 DimensionVector req_block_sizes{block_size_z, 1, 1}; in ComputeMappingSchemeAndReductionKind() local
3723 dims_in_elem, tile_size_y, tile_size_x, req_block_sizes, num_threads_y, in ComputeMappingSchemeAndReductionKind()
|