HomeSort by relevance Sort by last modified time
    Searched defs:lhs_index (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
dot_op_emitter.cc 765 llvm_ir::IrArray::Index lhs_index = EmitOperandArrayLoopNest( local
785 lhs_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue();
816 lhs_array_.EmitReadArrayElement(lhs_index, ir_builder_);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 505 llvm_ir::IrArray::Index lhs_index = EmitOperandArrayLoopNest( local
518 lhs_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue();
545 lhs_array.EmitReadArrayElement(lhs_index, &ir_builder_);
576 for (size_t dimension = 0; dimension < lhs_index.size(); ++dimension) {
578 target_index.push_back(lhs_index[dimension]);
    [all...]

Completed in 88 milliseconds