Home
last modified time | relevance | path

Searched defs:indvar (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.cc41 [&](llvm::Value* indvar, bool is_first_iteration) -> Status { in ForWithStatus()
Dllvm_loop.cc121 llvm::Value* indvar = indvar_; in Emit() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc352 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dwhile_test.cc1208 auto indvar = GetTupleElement(state, 0); in XLA_TEST_F() local
1247 auto indvar = Parameter(&builder, 0, while_shape, "state"); in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc959 /*step=*/1, [&](llvm::Value* indvar) { in EmitBatchDotOperation()