| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | bridge_pass.cc | 30 mlir::TFTPU::CreateTPUBridgePipeline);
37 mlir::TFTPU::CreateTPUBridgePipelineV1);
|
| D | bridge.h | 23 namespace TFTPU {
|
| D | tpu_cluster_cleanup_attributes.cc | 26 namespace TFTPU { namespace
|
| D | tpu_identity_pruning.cc | 32 namespace TFTPU { namespace
|
| D | outside_compiled_to_host_launch.cc | 26 namespace TFTPU { namespace
|
| D | tpu_host_computation_expansion.cc | 31 namespace TFTPU { namespace
|
| D | tpu_compile_op_replication_pass.cc | 25 namespace TFTPU { namespace
|
| D | tpu_colocate_composite_resource_ops.cc | 30 namespace TFTPU { namespace
|
| D | tpu_resource_read_for_write.cc | 31 namespace TFTPU { namespace
|
| D | tpu_reorder_replicate_and_partitioned_inputs.cc | 24 namespace TFTPU { namespace
|
| D | tpu_update_embedding_enqueue_op_inputs.cc | 35 namespace TFTPU { namespace
|
| D | tpu_parallel_execute_sink_resource_write.cc | 30 namespace TFTPU { namespace
|
| D | tpu_resource_partitioning.cc | 33 namespace TFTPU { namespace
|
| D | bridge.cc | 41 namespace TFTPU { namespace
|
| D | tpu_dynamic_padding_mapper.cc | 40 namespace TFTPU { namespace
|
| D | tf_passes.td | 212 let constructor = "TFTPU::CreateTPUClusterFormationPass()";
335 let constructor = "TFTPU::CreateTPUResourceReadForWritePass()";
391 let constructor = "TFTPU::CreateTPUExtractOutsideCompilationPass()";
543 let constructor = "TFTPU::CreateTPUReorderReplicateAndPartitionedInputsPass()";
593 let constructor = "TFTPU::CreateTPUResourceReadsWritesPartitioningPass()";
711 let constructor = "TFTPU::CreateTPURewritePass()";
|
| D | passes.h | 309 namespace TFTPU {
|
| D | tpu_device_propagation.cc | 34 namespace TFTPU { namespace
|
| D | tpu_sharding_identification_pass.cc | 41 namespace TFTPU { namespace
|
| D | tpu_dynamic_layout_pass.cc | 47 namespace TFTPU { namespace
|
| D | tpu_extract_head_tail_outside_compilation.cc | 46 namespace TFTPU { namespace
|
| D | tpu_variable_runtime_reformatting.cc | 56 namespace TFTPU { namespace
|
| D | tpu_extract_outside_compilation.cc | 46 namespace TFTPU { namespace
|
| D | tpu_cluster_formation.cc | 48 namespace TFTPU { namespace
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.cc | 136 mlir::TFTPU::TPUBridge(module, /*enable_logging=*/VLOG_IS_ON(1))); in Run()
180 mlir::TFTPU::TPUBridgeV1Compat(module, /*enable_logging=*/VLOG_IS_ON(1))); in Run()
|