Searched refs:autotuner_thresh (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUPartitionedCall.pbtxt | 59 name: "autotuner_thresh"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | TPUPartitionedCall.pbtxt | 59 name: "autotuner_thresh"
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 365 Node** compile_node, int64 autotuner_thresh);
559 int64 autotuner_thresh);
|
| D | distributed_tpu_rewrite_pass.cc | 2318 Node** compile_node, int64 autotuner_thresh) { in BuildCompileNode() argument
2420 proto.set_xla_fusion_autotuner_thresh(autotuner_thresh); in BuildCompileNode()
4218 int64 autotuner_thresh) { in RewriteTPUReplicateNode() argument
4326 dynamic_shape_nodes, graph, &compile_node, autotuner_thresh)); in RewriteTPUReplicateNode()
4474 int64 autotuner_thresh = options.session_options->config.experimental() in Run() local
4486 &tpu_replicate_device_names_mapping, autotuner_thresh)); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | shape_inference.mlir | 414 …%2 = "tf.TPUPartitionedCall"(%arg0, %arg1) {autotuner_thresh = 0 : i64, f = @tpu_partitioned_call_…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1282 DefaultValuedAttr<I64Attr, "0">:$autotuner_thresh
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4813 …argspec: "args=[\'args\', \'device_ordinal\', \'Tout\', \'f\', \'autotuner_thresh\', \'name\'], va…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4813 …argspec: "args=[\'args\', \'device_ordinal\', \'Tout\', \'f\', \'autotuner_thresh\', \'name\'], va…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 53899 name: "autotuner_thresh"
|