Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback.py84 tensor_debug_mode, argument
90 self._tensor_debug_mode = tensor_debug_mode
161 def tensor_debug_mode(self): member in _DumpingCallback
289 tensor_debug_mode): argument
311 if (tensor_debug_mode == debug_event_pb2.TensorDebugMode.FULL_TENSOR and
359 tensor_debug_mode = self._tensor_debug_mode
377 if tensor_debug_mode == debug_event_pb2.TensorDebugMode.NO_TENSOR:
398 op_type, tensor, debug_tensor, tensor_debug_mode))
399 elif tensor_debug_mode in (debug_event_pb2.TensorDebugMode.CURT_HEALTH,
405 tensor_debug_mode in (
[all …]
H A Ddumping_callback_test.py87 self.dump_root, tensor_debug_mode="NONSENSICAL")
98 tensor_debug_mode): argument
105 self.dump_root, tensor_debug_mode=tensor_debug_mode)
108 self.assertIn(tensor_debug_mode, log_messages[0])
121 def testPureEagerOpExecution(self, tensor_debug_mode): argument
130 self.dump_root, tensor_debug_mode=tensor_debug_mode)
172 debug_event_pb2.TensorDebugMode.keys()[execution.tensor_debug_mode],
173 tensor_debug_mode)
174 if tensor_debug_mode == "NO_TENSOR":
178 elif tensor_debug_mode == "CURT_HEALTH":
[all …]
H A Ddebug_v2_ops_test.py70 tensor_debug_mode=debug_event_pb2.TensorDebugMode.FULL_TENSOR,
79 tensor_debug_mode=debug_event_pb2.TensorDebugMode.FULL_TENSOR,
116 self.assertEqual(trace.tensor_debug_mode,
127 self.assertEqual(trace.tensor_debug_mode,
150 tensor_debug_mode=debug_event_pb2.TensorDebugMode.FULL_TENSOR,
175 self.assertEqual(trace.tensor_debug_mode,
200 tensor_debug_mode=debug_event_pb2.TensorDebugMode.FULL_TENSOR,
218 self.assertEqual(trace.tensor_debug_mode,
249 tensor_debug_mode=debug_event_pb2.TensorDebugMode.FULL_TENSOR,
290 tensor_debug_mode=(
[all …]
H A Ddebug_events_monitors_test.py68 def testOnExecutionIsCalled(self, tensor_debug_mode): argument
72 self.dump_root, tensor_debug_mode=tensor_debug_mode)
90 if tensor_debug_mode == "NO_TENSOR":
92 elif tensor_debug_mode == "CONCISE_HEALTH":
96 elif tensor_debug_mode == "FULL_HEALTH":
102 elif tensor_debug_mode == "FULL_TENSOR":
113 def testOnGraphExecutionTraceIsCalled(self, tensor_debug_mode): argument
116 self.dump_root, tensor_debug_mode=tensor_debug_mode)
143 if tensor_debug_mode == "CONCISE_HEALTH":
156 elif tensor_debug_mode == "FULL_HEALTH":
[all …]
H A Ddebug_events_monitors.py192 tensor_debug_mode, argument
218 assert tensor_debug_mode != debug_event_pb2.TensorDebugMode.FULL_TENSOR
221 if tensor_debug_mode == debug_event_pb2.TensorDebugMode.CURT_HEALTH:
230 elif tensor_debug_mode == debug_event_pb2.TensorDebugMode.CONCISE_HEALTH:
243 elif tensor_debug_mode == debug_event_pb2.TensorDebugMode.FULL_HEALTH:
263 if (execution.tensor_debug_mode ==
275 execution.tensor_debug_mode,
288 if (graph_execution_trace.tensor_debug_mode ==
299 graph_execution_trace.tensor_debug_mode,
H A Ddebug_events_reader.py425 tensor_debug_mode, argument
437 self._tensor_debug_mode = tensor_debug_mode
452 def tensor_debug_mode(self): member in Execution
480 "tensor_debug_mode": self.tensor_debug_mode,
793 tensor_debug_mode, argument
803 self._tensor_debug_mode = tensor_debug_mode
816 def tensor_debug_mode(self): member in GraphExecutionTrace
831 "tensor_debug_mode": self.tensor_debug_mode,
889 if (execution_proto.tensor_debug_mode ==
892 elif (execution_proto.tensor_debug_mode !=
[all …]
H A Dcheck_numerics_callback.py219 tensor_debug_mode=(
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DDebugIdentityV2.pbtxt37 name: "tensor_debug_mode"
89 name: "tensor_debug_mode"
148 name: "tensor_debug_mode"
H A DDebugNumericSummaryV2.pbtxt29 name: "tensor_debug_mode"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DDebugNumericSummaryV2.pbtxt16 name: "tensor_debug_mode"
58 name: "tensor_debug_mode"
H A DDebugIdentityV2.pbtxt37 name: "tensor_debug_mode"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_DebugIdentityV2.pbtxt32 name: "tensor_debug_mode"
48 The semantics of the input tensor depends on tensor_debug_mode. In typical
H A Dapi_def_DebugNumericSummaryV2.pbtxt11 name: "tensor_debug_mode"
84 depends on the tensor_debug_mode attribute.
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Ddebug_events_writer.cc277 const string& op_name, int32_t output_slot, int32_t tensor_debug_mode, in WriteGraphExecutionTrace() argument
287 if (tensor_debug_mode > 0) { in WriteGraphExecutionTrace()
288 trace->set_tensor_debug_mode(TensorDebugMode(tensor_debug_mode)); in WriteGraphExecutionTrace()
H A Ddebug_events_writer.h170 int32_t tensor_debug_mode,
/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…
H A Dtensorflow.raw_ops.pbtxt1125 …s=[\'input\', \'tfdbg_context_id\', \'op_name\', \'output_slot\', \'tensor_debug_mode\', \'debug_u…
1137 …argspec: "args=[\'input\', \'output_dtype\', \'tensor_debug_mode\', \'tensor_id\', \'name\'], vara…
/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…
H A Dtensorflow.raw_ops.pbtxt1125 …s=[\'input\', \'tfdbg_context_id\', \'op_name\', \'output_slot\', \'tensor_debug_mode\', \'debug_u…
1137 …argspec: "args=[\'input\', \'output_dtype\', \'tensor_debug_mode\', \'tensor_id\', \'name\'], vara…
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Ddebug_event.proto253 TensorDebugMode tensor_debug_mode = 6; field
291 TensorDebugMode tensor_debug_mode = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/examples/v2/
H A Ddebug_mnist_v2.py131 tensor_debug_mode=FLAGS.dump_tensor_debug_mode,
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Ddebug_event.pb.h2080 ::tensorflow::TensorDebugMode tensor_debug_mode() const; in Swap()
2299 ::tensorflow::TensorDebugMode tensor_debug_mode() const; in Swap()
4659 inline ::tensorflow::TensorDebugMode Execution::tensor_debug_mode() const { in tensor_debug_mode() function
4962 inline ::tensorflow::TensorDebugMode GraphExecutionTrace::tensor_debug_mode() const { in tensor_debug_mode() function
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt12942 name: "tensor_debug_mode"
13111 name: "tensor_debug_mode"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td3139 The semantics of the input tensor depends on tensor_debug_mode. In typical
3153 DefaultValuedAttr<I64Attr, "-1">:$tensor_debug_mode,