Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc203 UnaryVariantOpRegistry::VariantBinaryOpFn*
213 const VariantBinaryOpFn& add_fn) { in RegisterBinaryOpFn()
214 VariantBinaryOpFn* existing = GetBinaryOpFn(op, device, type_index); in RegisterBinaryOpFn()
219 binary_op_fns.insert(std::pair<FuncTuple<VariantBinaryOp>, VariantBinaryOpFn>( in RegisterBinaryOpFn()
Dvariant_op_registry.h66 VariantBinaryOpFn; typedef
125 const VariantBinaryOpFn& add_fn);
129 VariantBinaryOpFn* GetBinaryOpFn(VariantBinaryOp op, StringPiece device,
206 gtl::FlatMap<FuncTuple<VariantBinaryOp>, VariantBinaryOpFn, TupleHash>
297 UnaryVariantOpRegistry::VariantBinaryOpFn* binary_op_fn = in BinaryOpVariants()
Dvariant_op_registry_test.cc304 UnaryVariantOpRegistry::VariantBinaryOpFn f; in TEST()