Searched refs:select_op (Results 1 – 5 of 5) sorted by relevance
57 HloInstruction * select_op, in DoConditionalToSelect()59 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(conditional, select_op)); in DoConditionalToSelect()
1044 std::function<ReturnT(bool, ReturnT, ReturnT)> select_op =1052 ElementwiseTernaryOp(select, std::move(select_op)));
947 func @select_op(%arg0: i1) -> () {957 func @select_op(%arg1: vector<3xi1>) -> () {967 func @select_op(%arg1: vector<4xi1>) -> () {977 func @select_op(%arg1: vector<4xi1>) -> () {987 func @select_op(%arg1: vector<4xi1>) -> () {
1911 tensorflow::NodeDef* select_op = tensorflow_graph->add_node(); in ConvertSelectOperator() local1912 select_op->set_op("Select"); in ConvertSelectOperator()1913 select_op->set_name(src_op.outputs[0]); in ConvertSelectOperator()1915 *select_op->add_input() = src_op.inputs[0]; in ConvertSelectOperator()1916 *select_op->add_input() = src_op.inputs[1]; in ConvertSelectOperator()1917 *select_op->add_input() = src_op.inputs[2]; in ConvertSelectOperator()1920 (*select_op->mutable_attr())["T"].set_type(data_type); in ConvertSelectOperator()
96 "select_op.cc",