Searched refs:trace_op (Results 1 – 3 of 3) sorted by relevance
2 macro_rules! trace_op {16 trace_op!($name, true);20 trace_op!($name, false);
1951 trace_op = None1954 trace_op = self._save_tensor_value_to_cache_op(cache_idx,1963 trace_op = self._snapshot_tensor(processed_out_tensor)1982 trace_op = tpu_wrap_trace_fn(processed_out_tensor, tensor_name)1988 if trace_op:1990 tracing_ops.append(trace_op)1996 consumer_op._add_control_input(trace_op)
1270 tensorflow::profiler::TraceMe* trace_op = in Invoke() local1346 tflite::OnTfLiteOpInvokeEnd(trace_op); in Invoke()