Home
last modified time | relevance | path

Searched refs:step_marker_location (Results 1 – 17 of 17) 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 = "", padding_map = [], topology = "", device_…
44 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
67 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
85 // CHECK-DAG: step_marker_location = ""
94 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
118 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
144 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
166 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
180 // CHECK-DAG: step_marker_location = ""
195 …}) {num_cores_per_replica = 1, step_marker_location = "", padding_map = [], topology = "", device_…
[all …]
Dtpu_rewrite.mlir8 …cate = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = "", paddi…
23 …cate = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = "", paddi…
39 …cluster_func"() {_tpu_replicate = "cluster0", func = @empty_func, step_marker_location = "STEP_MAR…
54 …ate = "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 …cate = "cluster0", func = @empty_func, num_cores_per_replica = 1, step_marker_location = 1, paddin…
94 // Tests `tf_device.cluster_func` with unparsable `step_marker_location` attribute.
[all …]
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>…
Dgraph_pruning_preserve_ops.mlir12 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
Dtpu_space_to_depth_pass.mlir24 …A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], padding_map = [], step_marker_location = "STEP_MAR…
106 …A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], padding_map = [], step_marker_location = "STEP_MAR…
180 …A\01\01\22\01\00", "\08\01\1A\01\01\22\01\00"], padding_map = [], step_marker_location = "STEP_MAR…
Dtpu_cluster_formation.mlir394 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
419 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
/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"
/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.cc152 auto step_marker_location = in SetMetadataProtoStepMarkerLocation() local
154 if (!step_marker_location) in SetMetadataProtoStepMarkerLocation()
161 if (!step_marker_location.getValue().empty() && in SetMetadataProtoStepMarkerLocation()
163 std::string(step_marker_location.getValue()), &location)) in SetMetadataProtoStepMarkerLocation()
166 step_marker_location.getValue())); in SetMetadataProtoStepMarkerLocation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir12 …_per_replica = 1 : i64, num_replicas = 1 : i64, padding_map = [], step_marker_location = "STEP_MAR…
/external/tensorflow/tensorflow/core/protobuf/tpu/
Dcompile_metadata.proto103 xla.DebugOptions.StepMarkerLocation step_marker_location = 12; field
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc316 debug_options.set_xla_step_marker_location(metadata.step_marker_location()); in CreateHloModules()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops.td840 DefaultValuedAttr<StrAttr, "STEP_MARK_AT_ENTRY">:$step_marker_location,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt4817 … \'computation_shape\', \'host_compute_core\', \'padding_map\', \'step_marker_location\', \'allow_…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt4817 … \'computation_shape\', \'host_compute_core\', \'padding_map\', \'step_marker_location\', \'allow_…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt54025 name: "step_marker_location"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt81560 name: "step_marker_location"