Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2155 llvm::LoadInst* dyn_dim_size = b_.CreateLoad(source_buffer, "dyn_dim_size"); in HandleSliceToDynamic() local
2159 b_.CreateStore(dyn_dim_size, in HandleSliceToDynamic()
2161 dynamic_dims.push_back(b_.CreateIntCast(dyn_dim_size, b_.getInt64Ty(), in HandleSliceToDynamic()
2221 llvm::Value* dyn_dim_size = b_.CreateLoad( in HandlePadToStatic() local
2224 b_.CreateStore(dyn_dim_size, in HandlePadToStatic()
2227 dynamic_dims.push_back(b_.CreateIntCast(dyn_dim_size, b_.getInt64Ty(), in HandlePadToStatic()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc861 llvm::Value* dyn_dim_size = b_.CreateLoad( in EmitPadToStaticFromMlir() local
864 dynamic_dims.push_back(dyn_dim_size); in EmitPadToStaticFromMlir()
983 llvm::LoadInst* dyn_dim_size = b_.CreateLoad(source_buffer, "dyn_dim_size"); in EmitSliceToDynamicFromMlir() local
984 dynamic_dims.push_back(dyn_dim_size); in EmitSliceToDynamicFromMlir()