Searched refs:enable_whole_mesh_compilations (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_configuration_rewrite_pass.cc | 57 bool enable_whole_mesh_compilations, in AddConfigurationNode() argument
64 AddNodeAttr("enable_whole_mesh_compilations", enable_whole_mesh_compilations, in AddConfigurationNode()
79 bool enable_whole_mesh_compilations, in AddHostConfigNode() argument
85 AddNodeAttr("enable_whole_mesh_compilations", enable_whole_mesh_compilations, in AddHostConfigNode()
275 bool enable_whole_mesh_compilations = false; in Run() local
280 &enable_whole_mesh_compilations); in Run()
295 enable_whole_mesh_compilations, &configuration_node)); in Run()
311 enable_whole_mesh_compilations, &host_configuration_node)); in Run()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ConfigureDistributedTPU.pbtxt | 58 name: "enable_whole_mesh_compilations"
94 name: "enable_whole_mesh_compilations"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ConfigureDistributedTPU.pbtxt | 58 name: "enable_whole_mesh_compilations"
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 249 bool enable_whole_mesh_compilations; member
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_configuration_ops.cc | 303 params.enable_whole_mesh_compilations = enable_whole_mesh_compilations_; in Compute()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 837 …ng_config\', \'tpu_embedding_config\', \'is_global_init\', \'enable_whole_mesh_compilations\', \'c…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 837 …ng_config\', \'tpu_embedding_config\', \'is_global_init\', \'enable_whole_mesh_compilations\', \'c…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 2097 DefaultValuedAttr<BoolAttr, "false">:$enable_whole_mesh_compilations,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 8392 name: "enable_whole_mesh_compilations"
|