Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc143 const auto kDimZ = KernelMappingScheme::DimZ; variable
4373 llvm::Value* x_block_offset = b->CreateMul(index[kDimZ], x_dim_size); in GetUntransposedOutputLinearAddress()
4764 llvm::Value* starting_tile_index_for_dim = tile_origin[kDimZ]; in EmitTilingKernel()
4768 llvm::Value* last_block_for_dim = constant(dims_in_blocks[kDimZ] - 1); in EmitTilingKernel()
4770 constant(dims_in_elems[kDimZ] - in EmitTilingKernel()
4771 (dims_in_blocks[kDimZ] - 1) * mapping_scheme.GetTileSizeZ()); in EmitTilingKernel()
4781 block_dim_induction_var, kDimZ, &b_); in EmitTilingKernel()