Searched defs:float_input (Results 1 – 4 of 4) sorted by relevance
32 std::shared_ptr<Tensor> float_input; in Compute() local
36 std::shared_ptr<Tensor> float_input; in Compute() local
92 …bool float_input = input_type_value == kNumberTypeFloat16 || input_type_value == kNumberTypeFloat3… in MatchRemoveCastOpRule() local
254 explicit TensorIndex(float float_input) : float_(float_input), type_(TensorIndexType::Float) {} in TensorIndex()255 explicit TensorIndex(const py::float_ &float_input) : TensorIndex(float_input.cast<float>()) {} in TensorIndex()