Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc534 std::vector<llvm::Value*> rhs_multi_index = in HandleDot() local
542 rhs_multi_index[i] = lhs_multi_index[i]; in HandleDot()
554 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()
582 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_array.GetShape(), in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc435 std::vector<llvm::Value*> rhs_multi_index = in EmitNaiveLlvmIrGemm() local
460 rhs_multi_index[rhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
461 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_shape, in EmitNaiveLlvmIrGemm()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2152 std::vector<llvm::Value*> lhs_multi_index, rhs_multi_index; in EmitElementalDot() local
2163 rhs_multi_index.push_back( in EmitElementalDot()
2167 rhs_multi_index.push_back(dot_result_index[lhs_dims - 1 + i]); in EmitElementalDot()
2169 rhs_multi_index.insert(rhs_multi_index.begin() + rhs_contracting_dim, in EmitElementalDot()
2171 IrArray::Index rhs_index(rhs_multi_index, hlo->operand(1)->shape(), in EmitElementalDot()