Searched refs:VariantBinaryOp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 47 enum VariantBinaryOp { enum
123 void RegisterBinaryOpFn(VariantBinaryOp op, const string& device,
129 VariantBinaryOpFn* GetBinaryOpFn(VariantBinaryOp op, StringPiece device,
206 gtl::FlatMap<FuncTuple<VariantBinaryOp>, VariantBinaryOpFn, TupleHash>
288 Status BinaryOpVariants(OpKernelContext* ctx, VariantBinaryOp op, in BinaryOpVariants()
402 UnaryVariantBinaryOpRegistration(VariantBinaryOp op, const string& device, in UnaryVariantBinaryOpRegistration()
|
| D | variant_op_registry.cc | 204 UnaryVariantOpRegistry::GetBinaryOpFn(VariantBinaryOp op, StringPiece device, in GetBinaryOpFn()
212 VariantBinaryOp op, const string& device, const TypeIndex& type_index, in RegisterBinaryOpFn()
219 binary_op_fns.insert(std::pair<FuncTuple<VariantBinaryOp>, VariantBinaryOpFn>( in RegisterBinaryOpFn()
|