Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for REGISTER_OP (0.23 sec)

  1. tensorflow/c/experimental/ops/README.md

    category names correspond to generated source file names, and should be
    consistent with the original source files registering each operator. For example
    since `REGISTER_OP("MatMul")` appears in ***core/math_ops.cc***, the "MatMul"
    operator in the script should be in the "math" category, and it will be
    generated in the output file `c/experimental/ops/math_ops.cc`.
    
    Plain Text
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Jul 28 17:21:01 GMT 2021
    - 993 bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

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

    #     visibility = ["//visibility:public"],
    # )
    # copybara:uncomment_end
    
    # 'enable_registration_v2' opts-in to a different implementation of op and
    # kernel registration - REGISTER_OP, REGISTER_KERNEL_BUILDER, etc.
    #
    # This setting is currently experimental. The 'v2' implementation does _not_
    # correspond to a particular, finalized design; rather, it relates to
    # developing one.
    #
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 09 18:15:11 GMT 2024
    - 53.4K bytes
    - Viewed (8)
  4. tensorflow/c/eager/c_api_test.cc

    }
    TEST(CAPI, ExecuteWithTracing) { ExecuteWithTracing(false); }
    TEST(CAPI, ExecuteWithTracingAsync) { ExecuteWithTracing(true); }
    
    REGISTER_OP("TestNonCommUnavailable")
        .Output("out: string")
        .Doc(R"doc(Test non-communication op throwing Unavailable error.)doc");
    
    REGISTER_OP("TestCommUnavailable")
        .Output("out: string")
        .SetIsDistributedCommunication()
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  5. tensorflow/c/c_api_function_test.cc

                                 /*opts=*/nullptr, description, s.get());
      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,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  6. tensorflow/c/c_api.h

    // The calls to TF_AddInput and TF_AddInputList must match (in number,
    // order, and type) the op declaration.  For example, the "Concat" op
    // has registration:
    //   REGISTER_OP("Concat")
    //       .Input("concat_dim: int32")
    //       .Input("values: N * T")
    //       .Output("output: T")
    //       .Attr("N: int >= 2")
    //       .Attr("T: type");
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  7. 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.
    
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
Back to top