Home
last modified time | relevance | path

Searched refs:is_layout_sensitive (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cse.h32 explicit HloCSE(bool is_layout_sensitive,
34 : is_layout_sensitive_(is_layout_sensitive), in is_layout_sensitive_() argument
Dalgebraic_simplifier.h51 void set_is_layout_sensitive(bool is_layout_sensitive) { in set_is_layout_sensitive() argument
52 is_layout_sensitive_ = is_layout_sensitive; in set_is_layout_sensitive()
55 bool is_layout_sensitive() const { return is_layout_sensitive_; } in is_layout_sensitive() function
Dhlo_computation.h314 bool Equal(const HloComputation& other, bool is_layout_sensitive) const { in Equal() argument
315 return EqualInternal(other, is_layout_sensitive, in Equal()
323 bool is_layout_sensitive) const { in EqualIgnoringChannelIdValues() argument
324 return EqualInternal(other, is_layout_sensitive, in EqualIgnoringChannelIdValues()
506 bool EqualInternal(const HloComputation& other, bool is_layout_sensitive,
Dconditional_code_motion.h96 explicit ConditionalCodeMotion(bool is_layout_sensitive,
99 : is_layout_sensitive_(is_layout_sensitive), in is_layout_sensitive_() argument
Dhlo_cse.cc47 bool is_layout_sensitive) { in CombineConstants() argument
66 if (!is_layout_sensitive) { in CombineConstants()
Dconditional_code_motion.cc207 bool is_layout_sensitive) { in InstructionWithinBranchIdentical() argument
210 bool eq_operands = is_layout_sensitive in InstructionWithinBranchIdentical()
234 *instruction, eq_operand, eq_computations, is_layout_sensitive); in InstructionWithinBranchIdentical()
244 is_layout_sensitive); in InstructionWithinBranchIdentical()
311 bool is_layout_sensitive) { in FindSpecialConverts() argument
326 is_layout_sensitive in FindSpecialConverts()
391 bool is_layout_sensitive) { in ConvertSpecialMove() argument
425 old_root, branch_count, conditional, is_layout_sensitive); in ConvertSpecialMove()
878 HloInstruction* conditional, bool is_layout_sensitive, in GroupConnectedBoundaries() argument
884 is_layout_sensitive_(is_layout_sensitive), in GroupConnectedBoundaries()
Dhlo_computation.cc849 bool is_layout_sensitive, in EqualInternal() argument
874 return a->EqualInternal(*b, is_layout_sensitive, in EqualInternal()
880 *pair.second, operands_eq, comp_eq, is_layout_sensitive) in EqualInternal()
882 is_layout_sensitive); in EqualInternal()
Dalgebraic_simplifier.cc219 if (!options.is_layout_sensitive()) { in BitcastingOperandOfReshapeOrCopyChain()
564 if (options_.is_layout_sensitive()) { in SameShape()
926 if (options_.is_layout_sensitive()) { in HandleAdd()
1240 if (options_.is_layout_sensitive()) { in HandleConcatenate()
2101 if (options_.is_layout_sensitive()) { in OptimizeDotOfReorderContractingDims()
2309 if (options_.is_layout_sensitive()) { in HandleDot()
3149 if (options_.is_layout_sensitive()) { in HandleBroadcast()
3515 if (!options_.is_layout_sensitive()) { in HandlePad()
4298 if (!options_.is_layout_sensitive()) { in HandleSlice()
4564 if (options_.is_layout_sensitive()) { in HandleReduce()
[all …]