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
113 void RegisterUnaryOpFn(VariantUnaryOp op, const string& device,
119 VariantUnaryOpFn* GetUnaryOpFn(VariantUnaryOp op, StringPiece device,
204 gtl::FlatMap<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn, TupleHash>
265 Status UnaryOpVariant(OpKernelContext* ctx, VariantUnaryOp op, const Variant& v, in UnaryOpVariant()
373 UnaryVariantUnaryOpRegistration(VariantUnaryOp op, const string& device, in UnaryVariantUnaryOpRegistration()
|
| D | variant_op_registry.cc | 162 VariantUnaryOp op, StringPiece device, const TypeIndex& type_index) { in GetUnaryOpFn()
169 VariantUnaryOp op, const string& device, const TypeIndex& type_index, in RegisterUnaryOpFn()
176 unary_op_fns.insert(std::pair<FuncTuple<VariantUnaryOp>, VariantUnaryOpFn>( in RegisterUnaryOpFn()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_common.h | 254 template <typename Device, VariantUnaryOp OpEnum>
|