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

  /external/tensorflow/tensorflow/core/framework/
variant_op_registry.cc 48 const VariantShapeFn& shape_fn) {
55 GetPersistentStringPiece(type_name), shape_fn));
62 UnaryVariantOpRegistry::VariantShapeFn* shape_fn = local
64 if (shape_fn == nullptr) {
69 return (*shape_fn)(v, shape);
variant_op_registry.h 93 void RegisterShapeFn(const string& type_name, const VariantShapeFn& shape_fn);
327 const LocalVariantShapeFn& shape_fn) {
330 [type_name, shape_fn](const Variant& v, TensorShape* s) -> Status {
337 return shape_fn(*t, s);
variant_op_registry_test.cc 124 auto* shape_fn = local
126 EXPECT_NE(shape_fn, nullptr);
131 Status s0 = (*shape_fn)(v, &shape);
137 TF_EXPECT_OK((*shape_fn)(v, &shape));

Completed in 599 milliseconds