Searched refs:lhs_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 579 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_array.GetShape(), in HandleDot() local
581 llvm::Value* lhs_element = lhs_array.EmitReadArrayElement(lhs_index, &b_); in HandleDot()
610 for (size_t dimension = 0; dimension < lhs_index.size(); ++dimension) { in HandleDot()
612 target_multi_index.push_back(lhs_index[dimension]); in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 458 llvm_ir::IrArray::Index lhs_index(lhs_multi_index, lhs_shape, in EmitNaiveLlvmIrGemm() local
490 llvm::Value* lhs_element = lhs_array_.EmitReadArrayElement(lhs_index, b_); in EmitNaiveLlvmIrGemm()
526 for (int dimension = 0; dimension < lhs_index.size(); ++dimension) { in EmitNaiveLlvmIrGemm()
528 target_multi_index.push_back(lhs_index[dimension]); in EmitNaiveLlvmIrGemm()
538 target_multi_index, target_array_.GetShape(), lhs_index.GetType()); in EmitNaiveLlvmIrGemm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1296 DimensionVector lhs_index(lhs_rank); in HandleDotSlowPath()
1311 {&lhs_index[dnums.lhs_batch_dimensions(i)], in HandleDotSlowPath()
1319 result_index_locations.push_back({&lhs_index[i], nullptr}); in HandleDotSlowPath()
1338 {&lhs_index[lhs_dnum], &rhs_index[rhs_dnum]}); in HandleDotSlowPath()
1365 static_cast<ElementwiseT>(lhs_literal.Get<ReturnT>(lhs_index)) * in HandleDotSlowPath()
|
| D | elemental_ir_emitter.cc | 2158 IrArray::Index lhs_index(lhs_multi_index, hlo->operand(0)->shape(), in EmitElementalDot() local
2175 TF_ASSIGN_OR_RETURN(llvm::Value * lhs_value, lhs_generator(lhs_index)); in EmitElementalDot()
|