Searched refs:HloOp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_window_utils.cc | 35 class HloOp { class
37 HloOp() = default;
38 explicit HloOp(HloInstruction* inst) : inst_(inst) {} in HloOp() function in xla::__anon0a39b4b30111::HloOp
50 HloOp BinaryOp(HloOp x, HloOp y, HloOpcode opcode, in BinaryOp()
55 return HloOp(x.get()->parent()->AddInstruction( in BinaryOp()
59 HloOp operator+(HloOp x, HloOp y) { return BinaryOp(x, y, HloOpcode::kAdd); } in operator +()
61 HloOp operator-(HloOp x, HloOp y) { in operator -()
65 HloOp operator*(HloOp x, HloOp y) { in operator *()
69 HloOp operator/(HloOp x, HloOp y) { return BinaryOp(x, y, HloOpcode::kDivide); } in operator /()
71 HloOp Maximum(HloOp x, HloOp y, const std::string& name = "") { in Maximum()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/ |
| D | map_chlo_to_hlo_op.h | 62 #define POPULATE_BCAST(ChloOp, HloOp) \ in PopulateForBroadcastingBinaryOp() argument
64 Pattern<ChloOp, HloOp, HloBinaryElementwiseAdaptor<ChloOp, HloOp>>>( \ in PopulateForBroadcastingBinaryOp()
|