Searched refs:lhs_reduction_dimension (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 510 const int64 lhs_reduction_dimension = dnums.lhs_contracting_dimensions(0); in HandleDot() local
515 CHECK_NE(lhs_reduction_dimension, batch_dim); in HandleDot()
520 TF_RET_CHECK(lhs_shape.dimensions(lhs_reduction_dimension) == in HandleDot()
522 << "lhs_shape.dimensions(" << lhs_reduction_dimension in HandleDot()
523 << ") = " << lhs_shape.dimensions(lhs_reduction_dimension) in HandleDot()
533 lhs_array, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in HandleDot()
548 /*end_index=*/lhs_shape.dimensions(lhs_reduction_dimension), in HandleDot()
553 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in HandleDot()
611 if (dimension != lhs_reduction_dimension) { in HandleDot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 416 int64 lhs_reduction_dimension = dim_nums.lhs_contracting_dimensions(0); in EmitNaiveLlvmIrGemm() local
420 CHECK_EQ(lhs_shape.dimensions(lhs_reduction_dimension), in EmitNaiveLlvmIrGemm()
424 lhs_reduction_dimension == LayoutUtil::Minor(lhs_shape.layout(), 0); in EmitNaiveLlvmIrGemm()
434 lhs_array_, /*dimension_to_skip=*/lhs_reduction_dimension, "lhs"); in EmitNaiveLlvmIrGemm()
448 0, lhs_shape.dimensions(lhs_reduction_dimension), "reduction", in EmitNaiveLlvmIrGemm()
457 lhs_multi_index[lhs_reduction_dimension] = reduction_loop->GetIndVarValue(); in EmitNaiveLlvmIrGemm()
527 if (dimension != lhs_reduction_dimension) { in EmitNaiveLlvmIrGemm()
|