HomeSort by relevance Sort by last modified time
    Searched refs:RegisterShapeFn (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/framework/
variant_op_registry.h 93 void RegisterShapeFn(const string& type_name, const VariantShapeFn& shape_fn);
328 UnaryVariantOpRegistry::Global()->RegisterShapeFn(
variant_op_registry_test.cc 145 registry.RegisterShapeFn(kTypeName, f);
146 EXPECT_DEATH(registry.RegisterShapeFn(kTypeName, f),
variant_op_registry.cc 47 void UnaryVariantOpRegistry::RegisterShapeFn(const string& type_name,

Completed in 242 milliseconds