- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TensorListLength (0.32 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir
} // ----- // CHECK-LABEL: listLength func.func @listLength(%arg0: tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<i32> { %0 = "tf.TensorListLength"(%arg0) : (tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<i32> // CHECK: %0 = "tfl.custom"(%arg0) <{custom_code = "TensorListLength", custom_option = #tfl<const_bytes : "0x">}> : (tensor<!tf_type.variant<tensor<*xi32>>>) -> tensor<i32> func.return %0 : tensor<i32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
const llvm::SmallDenseMap<Value, SizeInfo>& buffer_to_size) { auto it = buffer_to_size.find(length.getInputHandle()); if (it == buffer_to_size.end()) { length.emitOpError("found tf.TensorListLength on unknown TensorList."); return failure(); } OpBuilder builder(length); if (it->getSecond().fixed) { auto dim = cutil::CreateScalarConst(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
TF_Tensor:$item ); TF_DerivedResultTypeAttr element_dtype = TF_DerivedResultTypeAttr<0>; let hasCanonicalizer = 1; } def TF_TensorListLengthOp : TF_Op<"TensorListLength", [Pure]> { let summary = "Returns the number of tensors in the input tensor list."; let description = [{ input_handle: the input list length: the number of tensors in the list }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)