Home
last modified time | relevance | path

Searched refs:use_tpu (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUReplicateMetadata.pbtxt23 name: "use_tpu"
84 name: "use_tpu"
152 name: "use_tpu"
227 name: "use_tpu"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTPUReplicateMetadata.pbtxt23 name: "use_tpu"
84 name: "use_tpu"
152 name: "use_tpu"
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py113 use_tpu=True,
163 use_tpu=True,
Dtpu.py1180 use_tpu: bool = True,
1401 num_replicas=num_replicas, use_tpu=use_tpu, **metadata_kwargs)
1521 if use_tpu:
Dtpu_embedding_v2_test.py1209 def test_optimizer_with_slot_creation_fn(self, use_tpu): argument
1222 if use_tpu:
1235 if use_tpu:
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUReplicateMetadata.pbtxt23 name: "use_tpu"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.estimator.tpu.-t-p-u-estimator.pbtxt24 …argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'use_tpu\', \'trai…
Dtensorflow.raw_ops.pbtxt4817 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dgraph_pruning_preserve_ops.mlir12 …TEP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = true, use_tpu = true} : () -> ()
Dtpu_space_to_depth_pass.mlir24 …pology = "\0A\04\01\01\01\02\10\01\18\02\22\08\00\00\00\00\00\00\00\01", use_tpu = true} : (tensor…
106 …EP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = false, use_tpu = true} : (tensor…
180 …EP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = false, use_tpu = true} : (tensor…
Dtpu_cluster_formation.mlir394 …EP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = false, use_tpu = true} : () -> ()
419 …EP_MARK_AT_ENTRY", topology = "", use_spmd_for_xla_partitioning = false, use_tpu = true} : () -> ()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir12 …ng_map = [], step_marker_location = "STEP_MARK_AT_ENTRY", topology = "", use_tpu = true, use_spmd_…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-function-defs.pbtxt127 key: "use_tpu"
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2613 bool use_tpu; in BuildTPUReplicateOps() local
2614 TF_RETURN_IF_ERROR(GetNodeAttr(replicate->attrs(), "use_tpu", &use_tpu)); in BuildTPUReplicateOps()
2615 if (!use_tpu) { in BuildTPUReplicateOps()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td835 DefaultValuedAttr<BoolAttr, "true">:$use_tpu,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4817 …argspec: "args=[\'num_replicas\', \'num_cores_per_replica\', \'topology\', \'use_tpu\', \'device_a…
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt81460 name: "use_tpu"
81521 name: "use_tpu"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53986 name: "use_tpu"