Home
last modified time | relevance | path

Searched refs:GetNumberOfBlocks (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.h124 int64 GetNumberOfBlocks() const { in GetNumberOfBlocks() function
Dkernel_tiling.cc162 llvm_ir::AddRangeMetadata(0, GetNumberOfBlocks(), in EmitBlockIndex()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3077 LaunchDimensions(mapping_scheme->GetNumberOfBlocks(), in EmitKernel()