MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CPUResourceOutput

Function CPUResourceOutput

tensorflow/core/common_runtime/placer_test.cc:2665–2680  ·  view source on GitHub ↗

Function that returns a resource that can be produced on CPU only.

Source from the content-addressed store, hash-verified

2663
2664// Function that returns a resource that can be produced on CPU only.
2665FunctionDef CPUResourceOutput() {
2666 return FDH::Create(
2667 // Name
2668 "CPUResourceOutput",
2669 // Args
2670 {"x: float"},
2671 // Return values
2672 {"ds: resource", "x_out: float"},
2673 // Attr def
2674 {},
2675 // Nodes
2676 {
2677 {{"make_ds"}, "CreateDatasetCPU", {}},
2678 },
2679 {{"ds", "make_ds:o:0"}, {"x_out", "x"}});
2680}
2681
2682TEST_F(NestedPlacerTest, DeepDeviceConstraintsPropagated) {
2683 /*

Callers 1

TEST_FFunction · 0.85

Calls 1

CreateFunction · 0.50

Tested by

no test coverage detected