- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GeneratorDatasetRegion (0.56 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/region-control-flow-to-functional.mlir
// CHECK: "tf.GeneratorDataset" // CHECK-DAG: @init // CHECK-DAG: @next // CHECK-DAG: @finalize // CHECK: return %0 = "tf.GeneratorDatasetRegion"(%arg0, %arg1, %arg2, %arg3) ({ ^bb0(%arg4: tensor<4xf32>): %1 = func.call @init(%arg4) : (tensor<4xf32>) -> tensor<7xf32> "tf.Yield"(%1) : (tensor<7xf32>) -> () }, {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 02 11:15:34 UTC 2024 - 44.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
while_op.getResults().take_front(while_region.getNumResults()))) std::get<0>(it).replaceAllUsesWith(std::get<1>(it)); while_region.erase(); return success(); } // Transform GeneratorDatasetRegion to GeneratorDatasetOp. LogicalResult RegionControlFlowToFunctional::ConvertGeneratorDatasetOp( SymbolTableCollection& symbol_table, GeneratorDatasetRegionOp regional) { mlir::MLIRContext* ctx = regional.getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
build($_builder, $_state, resultTypes, operands, attributes); }]>]; let hasCanonicalizer = 1; } def TF_GeneratorDatasetRegionOp : TF_Op<"GeneratorDatasetRegion", [AttrSizedOperandSegments, DeclareOpInterfaceMethods<RegionBranchOpInterface, [ "areTypesCompatible", "getEntrySuccessorOperands", "getRegionInvocationBounds",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)