Home
last modified time | relevance | path

Searched refs:op_regex (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback_test.py894 def testOpRegex(self, op_regex): argument
899 op_regex=op_regex)
926 if op_regex == "AddV2":
932 elif op_regex == "Log":
973 op_regex): argument
978 op_regex=op_regex)
1002 if tensor_dtypes == [dtypes.float32] and not op_regex:
1007 elif tensor_dtypes == ["float32"] and op_regex == "Sum":
1011 elif tensor_dtypes == (dtypes.float32,) and op_regex == "(?!Sum)":
1015 elif tensor_dtypes == [dtypes.int32] and not op_regex:
[all …]
H A Ddumping_callback.py86 op_regex, argument
92 self._op_regex = op_regex
687 op_regex=None, argument
854 op_regex,
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/
H A Dtesting.py129 def assertGraphContains(self, op_regex, n): argument
133 if re.match(op_regex, node.name):
137 if re.match(op_regex, node_def.name):
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tests/
H A Dimport_quant_stats.mlir39 …%0:2 = "tfl.split"(%cst, %arg0) {num_splits = 2 : i32, name = "op_regex"} : (tensor<i32>, tensor<4…
40 loc(fused["skip1.cc":10:8, "op_regex", callsite("skip2" at "skip3.cc":10:8)])
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.debugging.experimental.pbtxt9 …argspec: "args=[\'dump_root\', \'tensor_debug_mode\', \'circular_buffer_size\', \'op_regex\', \'te…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.debugging.experimental.pbtxt9 …argspec: "args=[\'dump_root\', \'tensor_debug_mode\', \'circular_buffer_size\', \'op_regex\', \'te…