Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 3 of 3 for REGISTER_OP (0.05 seconds)

The search processing time has exceeded the limit. The displayed results may be partial.

  1. tensorflow/c/c_api_test.cc

      TF_DeleteTensor(fetchValues[1]);
    }
    
    // REGISTER_OP for CApiAttributesTest test cases.
    // Registers two ops, each with a single attribute called 'v'.
    // The attribute in one op will have a type 'type', the other
    // will have list(type).
    #define ATTR_TEST_REGISTER_OP(type)                           \
      REGISTER_OP("CApiAttributesTestOp" #type)                   \
    Created: Tue Dec 30 12:39:10 GMT 2025
    - Last Modified: Mon Nov 17 00:00:38 GMT 2025
    - 97K bytes
    - Click Count (0)
  2. 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.
    #
    Created: Tue Dec 30 12:39:10 GMT 2025
    - Last Modified: Wed Nov 12 19:21:56 GMT 2025
    - 53.1K bytes
    - Click Count (0)
  3. 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,
    Created: Tue Dec 30 12:39:10 GMT 2025
    - Last Modified: Mon Nov 17 00:00:38 GMT 2025
    - 63.6K bytes
    - Click Count (1)
Back to Top