- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _index_to_I64 (0.34 sec)
-
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) -
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)