Searched refs:CreateUnaryOp (Results 1 – 1 of 1) sorted by relevance
3311 NodeDef CreateUnaryOp(DataType tf_type) { in CreateUnaryOp() function3319 NodeDef CreateUnaryOp<ops::internal::LeakyRelu>(DataType tf_type) { in CreateUnaryOp() function3333 const NodeDef& node_def = CreateUnaryOp<ops::Relu>(tf_type_); in TEST_P()3347 op_map[name] = std::make_pair(CreateUnaryOp<op>, compute) in TEST_P()5616 const NodeDef node_def = CreateUnaryOp<ops::Neg>(tf_type_); in TEST_P()5627 std::make_pair(CreateUnaryOp<op>, static_cast<ValFunc>(compute)) in TEST_P()