Home
last modified time | relevance | path

Searched defs: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
2221 llvm::Value* dyn_dim_size = b_.CreateLoad( in HandlePadToStatic() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc861 llvm::Value* dyn_dim_size = b_.CreateLoad( in EmitPadToStaticFromMlir() local
983 llvm::LoadInst* dyn_dim_size = b_.CreateLoad(source_buffer, "dyn_dim_size"); in EmitSliceToDynamicFromMlir() local