Searched refs:RegisterBinaryOpFn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 308 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, kTypeIndex, f); in TEST()
309 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, in TEST()
313 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, kTypeIndex, f); in TEST()
314 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_GPU, in TEST()
|
| D | variant_op_registry.h | 123 void RegisterBinaryOpFn(VariantBinaryOp op, const string& device,
406 UnaryVariantOpRegistry::Global()->RegisterBinaryOpFn( in UnaryVariantBinaryOpRegistration()
|
| D | variant_op_registry.cc | 211 void UnaryVariantOpRegistry::RegisterBinaryOpFn( in RegisterBinaryOpFn() function in tensorflow::UnaryVariantOpRegistry
|