- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SetAttrValue (0.11 sec)
-
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) -
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) -
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) -
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)