Searched refs:rhs_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 582 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_array.GetShape(), in HandleDot() local
584 llvm::Value* rhs_element = rhs_array.EmitReadArrayElement(rhs_index, &b_); in HandleDot()
617 dimension < rhs_index.size(); ++dimension) { in HandleDot()
619 target_multi_index.push_back(rhs_index[dimension]); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 461 llvm_ir::IrArray::Index rhs_index(rhs_multi_index, rhs_shape, in EmitNaiveLlvmIrGemm() local
491 llvm::Value* rhs_element = rhs_array_.EmitReadArrayElement(rhs_index, b_); in EmitNaiveLlvmIrGemm()
531 for (int dimension = 0; dimension < rhs_index.size(); ++dimension) { in EmitNaiveLlvmIrGemm()
533 target_multi_index.push_back(rhs_index[dimension]); in EmitNaiveLlvmIrGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1297 DimensionVector rhs_index(rhs_rank); in HandleDotSlowPath()
1312 &rhs_index[dnums.rhs_batch_dimensions(i)]}); in HandleDotSlowPath()
1325 result_index_locations.push_back({&rhs_index[i], nullptr}); in HandleDotSlowPath()
1338 {&lhs_index[lhs_dnum], &rhs_index[rhs_dnum]}); in HandleDotSlowPath()
1366 static_cast<ElementwiseT>(rhs_literal.Get<ReturnT>(rhs_index)); in HandleDotSlowPath()
|
| D | elemental_ir_emitter.cc | 2171 IrArray::Index rhs_index(rhs_multi_index, hlo->operand(1)->shape(), in EmitElementalDot() local
2176 TF_ASSIGN_OR_RETURN(llvm::Value * rhs_value, rhs_generator(rhs_index)); in EmitElementalDot()
|