Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for TF_OpDefinitionBuilderAddAttr (0.32 sec)

  1. tensorflow/c/kernels/ops/bitcast.cc

      TF_OpDefinitionBuilderAddOutput(op_builder, "output: type");
      TF_OpDefinitionBuilderAddAttr(
          op_builder,
          "T: {bfloat16, half, float, double, int64, int32, uint8, uint16, "
          "uint32, uint64, int8, int16, complex64, complex128, qint8, quint8, "
          "qint16, quint16, qint32}");
      TF_OpDefinitionBuilderAddAttr(
          op_builder,
          "type: {bfloat16, half, float, double, int64, int32, uint8, uint16, "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 07:51:50 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  2. tensorflow/c/ops_test.cc

      TF_DeleteStatus(status);
    }
    
    TEST(OpsTest, AttributeAccessors) {
      TF_OpDefinitionBuilder* builder =
          TF_NewOpDefinitionBuilder("AttributeAccessorsOp");
      TF_OpDefinitionBuilderAddAttr(builder, "foo1: int >= 2");
      TF_OpDefinitionBuilderAddAttr(builder, "foo2: string=\"my string\"");
      TF_OpDefinitionBuilderSetIsCommutative(builder, true);
      TF_OpDefinitionBuilderSetIsAggregate(builder, true);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 11 01:20:50 UTC 2021
    - 12.6K bytes
    - Viewed (0)
  3. tensorflow/c/kernels/ops/summary.cc

      TF_OpDefinitionBuilderAddInput(op_builder, "tags: string");
      TF_OpDefinitionBuilderAddInput(op_builder, "values: T");
      TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string");
      TF_OpDefinitionBuilderAddAttr(op_builder, "T: realnumbertype");
      TF_OpDefinitionBuilderSetShapeInferenceFunction(
          op_builder, &scalar_summary_shape_inference_fn);
    
      TF_RegisterOpDefinition(op_builder, status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Sep 18 21:04:53 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  4. tensorflow/c/kernels/ops/merge_summary.cc

          TF_NewOpDefinitionBuilder("MergeSummary");
      TF_OpDefinitionBuilderAddInput(op_builder, "inputs: N * string");
      TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string");
      TF_OpDefinitionBuilderAddAttr(op_builder, "N: int >= 1");
      TF_OpDefinitionBuilderSetShapeInferenceFunction(
          op_builder, &merge_summary_shape_inference_fn);
    
      TF_RegisterOpDefinition(op_builder, status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Sep 18 21:04:53 UTC 2023
    - 2K bytes
    - Viewed (0)
  5. tensorflow/c/kernels/ops/histogram_summary.cc

      TF_OpDefinitionBuilderAddInput(op_builder, "tag: string");
      TF_OpDefinitionBuilderAddInput(op_builder, "values: T");
      TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string");
      TF_OpDefinitionBuilderAddAttr(op_builder, "T: realnumbertype = DT_FLOAT");
      TF_OpDefinitionBuilderSetShapeInferenceFunction(
          op_builder, &histogram_summary_shape_inference_fn);
    
      TF_RegisterOpDefinition(op_builder, status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Sep 18 21:04:53 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  6. tensorflow/c/ops.h

    // corresponding functions, e.g.
    //
    // TF_OpDefinitionBuilderAddInput(builder, "input1: int32");
    // TF_OpDefinitionBuilderAddOutput(builder, "output1: int64");
    // TF_OpDefinitionBuilderAddAttr(builder, "attr: int32");
    //
    // The builder may then be registered with TensorFlow using the
    // TF_RegisterOpDefinition function. E.g.
    //
    // TF_Status* status = TF_NewStatus();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 16.3K bytes
    - Viewed (0)
  7. tensorflow/c/ops.cc

      }
    
    DEFINE_BUILDER_BOOL_SETTER(SetIsCommutative)
    DEFINE_BUILDER_BOOL_SETTER(SetIsAggregate)
    DEFINE_BUILDER_BOOL_SETTER(SetIsStateful)
    DEFINE_BUILDER_BOOL_SETTER(SetAllowsUninitializedInput)
    
    void TF_OpDefinitionBuilderAddAttr(TF_OpDefinitionBuilder* builder,
                                       const char* attr_spec) {
      reinterpret_cast<OpDefBuilder*>(builder)->Attr(attr_spec);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
Back to top