Searched refs:_emit_with_loc (Results 1 – 1 of 1) sorted by relevance
621 def _emit_with_loc(self, op_str, node=None): member in TFRGen656 self._emit_with_loc('\n{} = constant 0 : i64'.format(axis))659 self._emit_with_loc(' : (!tfr.tensor_list, i64) -> !tfr.tensor')667 self._emit_with_loc('\n{} = index_cast {} : index to i64'.format(676 self._emit_with_loc('\n{} = index_cast {} : i64 to index'.format(686 self._emit_with_loc(' : ({}) -> !tfr.tensor'.format(ty), node)703 self._emit_with_loc('\nscf.yield', node)705 self._emit_with_loc('\ntfr.return', node)722 self._emit_with_loc(733 self._emit_with_loc([all …]