- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TakeDataset (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
%cst = arith.constant dense<30> : tensor<i64> %cst_1 = arith.constant dense<-1> : tensor<i64> %cst_2 = arith.constant dense<165000> : tensor<i64> %0 = "tf.TakeDataset"(%arg0, %cst) {device = "", metadata = "", output_shapes = [#tf_type.shape<>], output_types = [!tf_type.string]} : (tensor<!tf_type.variant>, tensor<i64>) -> tensor<!tf_type.variant>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
}]; let arguments = (ins); let results = (outs TF_Int32Tensor:$device_ordinal ); } def TF_TakeDatasetOp : TF_Op<"TakeDataset", [Pure]> { let summary = [{ Creates a dataset that contains `count` elements from the `input_dataset`. }]; let arguments = (ins TF_VariantTensor:$input_dataset,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)