Home
last modified time | relevance | path

Searched refs:step_marker_location (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_extract_head_tail_outside_compilation.mlir23 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
44 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
67 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
85 // CHECK-DAG: step_marker_location = ""
93 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
117 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
143 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
165 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
201 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
213 // CHECK-DAG: step_marker_location = ""
[all …]
Dtpu_rewrite.mlir8 …info = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = "", topol…
23 …info = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = "", topol…
39 …type = "TPU", _replication_info = "cluster0", func = @empty_func, step_marker_location = "STEP_MAR…
54 …nfo = "cluster0", func = @empty_func, num_cores_per_replica = "", step_marker_location = "STEP_MAR…
64 // Tests `tf_device.cluster_func` with missing `step_marker_location` attribute.
68 // expected-error@+1 {{requires attribute 'step_marker_location'}}
79 // Tests `tf_device.cluster_func` with bad `step_marker_location` attribute.
83 // expected-error@+1 {{requires attribute 'step_marker_location'}}
84 …info = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = 1, topolo…
94 // Tests `tf_device.cluster_func` with unparsable `step_marker_location` attribute.
[all …]
Dcanonicalize_compile_and_replicate_attributes.mlir8 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
9 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
Dtpu_sharding_identification.mlir7 "tf_device.cluster_func"() {func = @empty_func, step_marker_location = ""} : () -> ()
24 …"tf_device.cluster_func"(%arg0) {func = @func_without_sharding, step_marker_location = ""} : (tens…
45 …"tf_device.cluster_func"(%arg0) {func = @func_without_sharding, step_marker_location = ""} : (tens…
66 …"tf_device.cluster_func"(%arg0) {func = @inputs_with_sharding_func, step_marker_location = ""} : (…
88 …"tf_device.cluster_func"(%arg0, %arg1) {func = @func_with_sharding, step_marker_location = ""} : (…
113 …r_func"(%arg0, %arg1) {func = @func_with_sharding_after_identity, step_marker_location = ""} : (te…
139 …c"(%arg0, %arg1) {func = @func_with_sharding_after_read_variable, step_marker_location = ""} : (te…
167 …uster_func"(%arg0, %arg1) {func = @func_with_sharding_after_cast, step_marker_location = ""} : (te…
194 …ster_func"(%arg0, %arg1) {func = @func_with_device_training_loop, step_marker_location = ""} : (te…
237 …"tf_device.cluster_func"(%arg0) {func = @cluster_func, step_marker_location = ""} : (tensor<*xi32>…
Dconvert_to_legacy_compile_and_replicate_attributes.mlir12 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
34 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
Dcannonicalize_ops_outside_compilation.mlir15 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
Dgraph_pruning_preserve_ops.mlir12 …re = [], num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
Dtpu_tail_with_tobool_op.mlir27 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
Dinlining.mlir91 …}) {num_cores_per_replica = 1, step_marker_location = "", topology = "", device_assignment = []} :…
Dtpu_space_to_depth_pass.mlir24 …ation = ["\08\01\1A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], step_marker_location = "STEP_MAR…
106 …\01\00", "\08\01\1A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], step_marker_location = "STEP_MAR…
180 …\01\00", "\08\01\1A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], step_marker_location = "STEP_MAR…
Dtpu_cluster_formation.mlir359 …re = [], num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
384 …re = [], num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
674 … = true, device_assignment = [], num_cores_per_replica = 1 : i32, step_marker_location = "", topol…
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUReplicateMetadata.pbtxt123 name: "step_marker_location"
191 name: "step_marker_location"
266 name: "step_marker_location"
348 name: "step_marker_location"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTPUReplicateMetadata.pbtxt123 name: "step_marker_location"
191 name: "step_marker_location"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc147 auto step_marker_location = in SetMetadataProtoStepMarkerLocation() local
149 if (!step_marker_location) in SetMetadataProtoStepMarkerLocation()
156 if (!step_marker_location.getValue().empty() && in SetMetadataProtoStepMarkerLocation()
158 std::string(step_marker_location.getValue()), &location)) in SetMetadataProtoStepMarkerLocation()
161 step_marker_location.getValue())); in SetMetadataProtoStepMarkerLocation()
Dtf_passes.td44 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
50 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
79 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
86 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
258 step_marker_location = ""} : (tensor<*xi32>) -> tensor<*xi32>
279 step_marker_location = ""} : (tensor<*xi32>) -> tensor<*xi32>
1803 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
1817 …signment = [], num_cores_per_replica = 1 : i64, padding_map = [], step_marker_location = "", topol…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir12 …tadata", num_cores_per_replica = 1 : i64, num_replicas = 1 : i64, step_marker_location = "STEP_MAR…
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dcompile_metadata.proto107 xla.DebugOptions.StepMarkerLocation step_marker_location = 12; field
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Dtest24.pbtxt226 key: "step_marker_location"
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc321 debug_options.set_xla_step_marker_location(metadata.step_marker_location()); in CreateHloModules()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4897 … \'computation_shape\', \'host_compute_core\', \'padding_map\', \'step_marker_location\', \'allow_…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4897 … \'computation_shape\', \'host_compute_core\', \'padding_map\', \'step_marker_location\', \'allow_…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt57802 name: "step_marker_location"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td17750 DefaultValuedAttr<StrAttr, "\"STEP_MARK_AT_ENTRY\"">:$step_marker_location,
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt81535 name: "step_marker_location"