Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SetAttrValue (0.23 sec)

  1. tensorflow/c/kernels/bitcast_op_test.cc

      Status status;
      NodeDef def;
      def.set_op("Bitcast");
      def.set_device(DEVICE_CPU);
    
      AttrValue typeAttr;
      SetAttrValue(input_tensor->dtype(), &typeAttr);
    
      AttrValue outTypeAttr;
      SetAttrValue(out_type, &outTypeAttr);
    
      (*def.mutable_attr())["T"] = typeAttr;
      (*def.mutable_attr())["type"] = outTypeAttr;
    
      def.add_input(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  2. tensorflow/c/kernels/summary_op_test.cc

      // Initialize node used to fetch OpKernel
      Status status;
      NodeDef def;
      def.set_op("ScalarSummary");
    
      def.set_device(DEVICE_CPU);
    
      AttrValue valuesTypeAttr;
      SetAttrValue(values->dtype(), &valuesTypeAttr);
      (*def.mutable_attr())["T"] = valuesTypeAttr;
    
      def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype())));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 6.7K bytes
    - Viewed (0)
Back to top