Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for _index_to_I64 (0.17 sec)

  1. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

        self._emit_with_loc(' : (!tfr.tensor_list, i64) -> !tfr.tensor')
        # load the op def of tf.Pack
        self._op_defs.lookup('Pack')
        return casted, TFRTypes.TENSOR
    
      def _index_to_I64(self, value, ty):
        if ty == TFRTypes.INDEX:
          casted = self._ssa_name('casted')
          self._emit_with_loc('\n{} = arith.index_cast {} : index to i64'.format(
              casted, value))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
Back to top