Home
last modified time | relevance | path

Searched refs:exempt_ops (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dinput_colocation_exemption_registry_test.cc28 auto exempt_ops = InputColocationExemptionRegistry::Global()->Get(); in TEST() local
29 EXPECT_EQ(exempt_ops.size(), 2); in TEST()
30 EXPECT_NE(exempt_ops.find("op 1"), exempt_ops.end()); in TEST()
31 EXPECT_NE(exempt_ops.find("op 2"), exempt_ops.end()); in TEST()
32 EXPECT_EQ(exempt_ops.find("op 3"), exempt_ops.end()); in TEST()
Dcolocation_graph.cc104 auto exempt_ops = InputColocationExemptionRegistry::Global()->Get(); in IsExemptFromResourceInputColocation() local
105 return exempt_ops.find(op_type) != exempt_ops.end(); in IsExemptFromResourceInputColocation()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_utils.cc63 const auto& exempt_ops = InputColocationExemptionRegistry::Global()->Get(); in IsColocationExempt() local
64 return exempt_ops.find(string(op_name)) != exempt_ops.end(); in IsColocationExempt()
Deager_operation.cc329 const auto& exempt_ops = InputColocationExemptionRegistry::Global()->Get(); in Reset() local
330 colocation_exempt_ = exempt_ops.find(op) != exempt_ops.end(); in Reset()