Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for SetShapeFn (0.24 sec)

  1. tensorflow/c/c_api_test.cc

          .Attr("v: " #type)                                      \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape); \
      REGISTER_OP("CApiAttributesTestOpList" #type)               \
          .Attr("v: list(" #type ")")                             \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape)
    ATTR_TEST_REGISTER_OP(string);
    ATTR_TEST_REGISTER_OP(int);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  2. tensorflow/c/c_api_function_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get());
      ASSERT_NE(*func, nullptr);
    }
    
    REGISTER_OP("CustomOp")
        .Output("output: float32")
        .Attr("index: int")
        .SetShapeFn(tensorflow::shape_inference::UnknownShape);
    
    void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
                                       const char* name, const char* placeholder,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  3. RELEASE.md

    *   C API gives access to inferred shapes with `TF_GraphGetTensorNumDims` and
        `TF_GraphGetTensorShape`.
    *   Shape functions for core ops have moved to C++ via
        `REGISTER_OP(...).SetShapeFn(...)`. Python shape inference RegisterShape
        calls use the C++ shape functions with `common_shapes.call_cpp_shape_fn`. A
        future release will remove `RegisterShape` from python.
    
    ## Bug Fixes and Other Changes
    Plain Text
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Apr 03 20:27:38 GMT 2024
    - 727.4K bytes
    - Viewed (8)
Back to top