Searched refs:select_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_to_select.cc | 61 HloInstruction * select_op, in DoConditionalToSelect()
63 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(conditional, select_op)); in DoConditionalToSelect()
|
| D | hlo_evaluator_typed_visitor.h | 860 std::function<ReturnT(bool, ReturnT, ReturnT)> select_op =
868 ElementwiseTernaryOp(select, std::move(select_op)));
|
| /external/ComputeLibrary/src/cpu/kernels/select/generic/neon/ |
| D | impl.cpp | 38 void select_op(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const Win… in select_op() function
82 …select_op<ScalarType, VectorType>(cond, in1, in2, out, window, window_step_x, window_start_x, wind… in select_op_8()
96 …select_op<ScalarType, VectorType>(cond, in1, in2, out, window, window_step_x, window_start_x, wind… in select_op_16()
110 …select_op<ScalarType, VectorType>(cond, in1, in2, out, window, window_step_x, window_start_x, wind… in select_op_32()
|
| D | impl.h | 49 void select_op(const ITensor *cond, const ITensor *in1, const ITensor *in2, ITensor *out, const Win…
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1916 tensorflow::NodeDef* select_op = tensorflow_graph->add_node(); in ConvertSelectOperator() local
1917 select_op->set_op("Select"); in ConvertSelectOperator()
1918 select_op->set_name(src_op.outputs[0]); in ConvertSelectOperator()
1920 *select_op->add_input() = src_op.inputs[0]; in ConvertSelectOperator()
1921 *select_op->add_input() = src_op.inputs[1]; in ConvertSelectOperator()
1922 *select_op->add_input() = src_op.inputs[2]; in ConvertSelectOperator()
1925 (*select_op->mutable_attr())["T"].set_type(data_type); in ConvertSelectOperator()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 114 "select_op.cc",
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 3114 auto select_op = CreateOpAndInfer<tosa::SelectOp>( in convertResizeOp() local
3119 rewriter, op->getLoc(), output_type, select_op.getResult()); in convertResizeOp()
|