Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for SetAttrValue (0.1 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_test.cc

              << i;
        }
        TF_DeleteStatus(status);
        return static_cast<void*>(s);
      };
    
      AttrValue v;
      std::string attr_in[] = {"bugs", "bunny", "duck"};
      SetAttrValue(gtl::ArraySlice<std::string>(attr_in, 3), &v);
      CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrStringList", v);
    }
    
    TEST_F(TestKernelAttr, Tensor) {
      struct TensorProtoHelpers {
       public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  3. 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)
  4. tensorflow/c/eager/c_api.cc

    // Annotate eager runtime construction context to the given `function_def` as
    // an attribute.
    void AnnotateEagerRuntimeConstructionContext(
        tensorflow::FunctionDef& function_def) {
      tensorflow::AttrValue value;
      SetAttrValue("kEagerRuntime", &value);
      (*function_def.mutable_attr())["_construction_context"] = value;
    }
    
    }  // namespace
    
    extern "C" {
    
    TFE_ContextOptions* TFE_NewContextOptions() { return new TFE_ContextOptions; }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top