Searched defs:is_min (Results 1 – 7 of 7) sorted by relevance
33 XlaArgMinMaxOp::XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min) in XlaArgMinMaxOp()
116 bool is_min) { in CreateMinMaxComputation()138 XlaOp ArgMinMax(XlaOp input, PrimitiveType output_type, int axis, bool is_min) { in ArgMinMax()
51 absl::Span<NativeT const> expected_output, int axis, bool is_min) { in TestArgMinMax()
176 constexpr bool is_min() const { in is_min() function358 bool is_min() const { return us_ == std::numeric_limits<int64_t>::min(); } in is_min() function
201 constexpr bool is_min() const { return *this == Min(); } in is_min() function433 constexpr bool is_min() const { return *this == Min(); } in is_min() function
422 constexpr bool is_min() const { return *this == Min(); } in is_min() function
3932 const bool is_min = index == 1; in AddQuantizedMatMulMinMaxOutConstNodes() local