Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for _index_to_I64 (0.34 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)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

            Value index_to_i64 = rewriter.create<arith::IndexCastOp>(
                op.getLoc(), rewriter.getI64Type(), slice_size_index);
            Value i64_to_tensor = rewriter.create<tensor::FromElementsOp>(
                op.getLoc(),
                tensorflow::GetTypeFromTFTensorShape({1}, rewriter.getI64Type()),
                index_to_i64);
            slice_sizes_values.push_back(i64_to_tensor);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top