Home
last modified time | relevance | path

Searched refs:VariantUnaryOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.h41 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()
Dvariant_op_registry.cc29 const char* VariantUnaryOpToString(VariantUnaryOp op) { in VariantUnaryOpToString()
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_common.h315 template <typename Device, VariantUnaryOp OpEnum>