Home
last modified time | relevance | path

Searched defs:use_tpu (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v2_optimizer_test.py133 def test_optimizer_with_slot_creation_fn(self, use_tpu): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmetrics.cc568 const std::string& num_cores_per_replica, const std::string& use_tpu, in UpdateTfMlirBridgeGraphAnalysisPerOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc2565 bool use_tpu; in BuildTPUReplicateOps() local