- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 57 for tensorlist (0.18 sec)
-
tensorflow/compiler/mlir/tfr/python/tfr_gen_test.py
CHECK: } CHECK-LABEL: tfr.func @tf__test_identity_n_op(%x: !tfr.tensor_list) -> (!tfr.tensor_list) { CHECK-NEXT: %[[raw_data:.*]] = tfr.quant_raw_data(%x) : (!tfr.tensor_list) -> (!tfr.tensor_list) CHECK: tfr.return %[[raw_data:.*]] : !tfr.tensor_list CHECK: } """ self._check_code(mlir_code, mlir_code_exp) if __name__ == '__main__':
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 13 16:33:28 UTC 2021 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/decompose.mlir
} // CHECK-LABEL: @tf__fused_n tfr.func @tf__fused_n( %arg0: !tfr.tensor, %arg1: !tfr.tensor_list, %arg2: index {tfr.name="A",tfr.default=1:index}) -> !tfr.tensor_list { %0 = tfr.call @tf__intermediate(%arg0) : (!tfr.tensor) -> !tfr.tensor %1 = tfr.get_element %arg1[%arg2] : (!tfr.tensor_list, index) -> !tfr.tensor %2 = tfr.call @tf__intermediate(%1) : (!tfr.tensor) -> !tfr.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
Example: ```mlir %2 = tfr.call @my_add(%0, %1) : (tfr.tensor, f32) -> tfr.tensor_list ``` Note that the operands of the `call` operation can only be with tfr.tensor, tfr.tensor_list, tfr.attr and mlir float and integer types. The results of the `call` operation can only be with tfr.tensor and tfr.tensor_list types. }]; let arguments = (ins FlatSymbolRefAttr:$callee,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/canonicalize.mlir
%5 = "tfr.build_list"(%3, %4) : (!tfr.tensor, !tfr.tensor) -> !tfr.tensor_list %6 = tfr.quant_raw_data(%5) : (!tfr.tensor_list) -> !tfr.tensor_list %7 = "tfr.cast"(%cst_1) : (tensor<i64>) -> !tfr.tensor %8 = tfr.call @tf__concat(%7, %6) : (!tfr.tensor, !tfr.tensor_list) -> !tfr.tensor func.return %8 : !tfr.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/resources/decomposition_lib.mlir
// OpDefs. tfr.func @tf__my_add_n(%values: !tfr.tensor_list, %n: i64 {tfr.name="N"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i64 %eq = arith.cmpi eq, %n, %cst : i64 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor %res = scf.if %eq -> !tfr.tensor { scf.yield %v1 : !tfr.tensor } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 13 16:33:28 UTC 2021 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/end2end.mlir
tfr.return %0 : !tfr.tensor } tfr.func @tf__my_pack(%values: !tfr.tensor_list, %n: i32 {tfr.name="N"}, %axis: i32 {tfr.name="axis"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i32 %eq = arith.cmpi eq, %n, %cst : i32 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/control_flow.mlir
tfr.func @tf__my_pack(%values: !tfr.tensor_list, %n: i32 {tfr.name="N"}, %axis: i32 {tfr.name="axis"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i32 %eq = arith.cmpi eq, %n, %cst : i32 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:58:25 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/raise_to_tf.mlir
tfr.func @tf__risc_same_(!tfr.tensor<T>) -> !tfr.tensor<T> attributes {T} tfr.func @tf__risc_concat_(!tfr.tensor_list<N, T>) -> !tfr.tensor<T> attributes {T, N} tfr.func @tf__risc_split_(!tfr.tensor<T>, i32 {tfr.name="N"}) -> !tfr.tensor_list<N, T> attributes {T, N} tfr.func @tf__risc_cast_(!tfr.tensor, !tfr.attr {tfr.name="K"}) -> !tfr.tensor<K> attributes {T, K}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx_test.cc
constexpr char tfr_raw_text[] = R"( tfr.func @tf__my_add_n(%values: !tfr.tensor_list, %n: i64 {tfr.name="N"}) -> !tfr.tensor { %index = arith.constant 0 : index %cst = arith.constant 1 : i64 %eq = arith.cmpi "eq", %n, %cst : i64 %v1 = tfr.get_element %values[%index] : (!tfr.tensor_list, index) -> !tfr.tensor %res = scf.if %eq -> !tfr.tensor { scf.yield %v1 : !tfr.tensor } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
input_used_attrs.append(used.begin(), used.end()); continue; } if (auto tensor_list = arg_type.dyn_cast<TFRTensorListType>()) { if (first_tensor_list == -1) { first_tensor_list = arg.index(); } last_tensor_list = arg.index(); auto used = tensor_list.getAttrKeys(); input_used_attrs.append(used.begin(), used.end()); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0)