Searched refs:VariantUnaryOp (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | variant_op_registry.h | 41 enum VariantUnaryOp { enum 47 const char* VariantUnaryOpToString(VariantUnaryOp op); 131 void RegisterUnaryOpFn(VariantUnaryOp op, const std::string& device, in RegisterUnaryOpFn() 139 unary_op_fns.insert(std::pair<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn>( in RegisterUnaryOpFn() 145 VariantUnaryOpFn* GetUnaryOpFn(VariantUnaryOp op, StringPiece device, in GetUnaryOpFn() 251 gtl::FlatMap<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn, TupleHash> 312 Status UnaryOpVariant(OpKernelContext* ctx, VariantUnaryOp op, const Variant& v, in UnaryOpVariant() 421 UnaryVariantUnaryOpRegistration(VariantUnaryOp op, const std::string& device, in UnaryVariantUnaryOpRegistration()
|
D | variant_op_registry.cc | 29 const char* VariantUnaryOpToString(VariantUnaryOp op) { in VariantUnaryOpToString()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cwise_ops_common.h | 315 template <typename Device, VariantUnaryOp OpEnum>
|