Home
last modified time | relevance | path

Searched refs:operand_set (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc72 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoDot() local
76 operand_set.push_back(i); in CanFoldOperandsIntoDot()
82 return transposable_gemm_operands(dot, operand_set); in CanFoldOperandsIntoDot()
93 TransposeFolding::OperandIndices operand_set; in CanFoldOperandsIntoConvolution() local
97 operand_set.push_back(i); in CanFoldOperandsIntoConvolution()
101 return transposable_conv_operands(convolution, operand_set); in CanFoldOperandsIntoConvolution()
Dhlo_dataflow_analysis.cc389 const InstructionValueSet& operand_set = in UpdateBitcastValueSet() local
392 if (!bitcast_defines_value_ && operand_set != bitcast_set) { in UpdateBitcastValueSet()
393 bitcast_set = operand_set; in UpdateBitcastValueSet()
402 const InstructionValueSet& operand_set = in UpdateSetDimensionSizeValueSet() local
406 if (operand_set != set_dimension_size_set) { in UpdateSetDimensionSizeValueSet()
407 set_dimension_size_set = operand_set; in UpdateSetDimensionSizeValueSet()
571 const InstructionValueSet& operand_set = in UpdateAddDependencyValueSet() local
575 if (operand_set != add_dependency_set) { in UpdateAddDependencyValueSet()
576 add_dependency_set = operand_set; in UpdateAddDependencyValueSet()