- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for MapDataset (0.09 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// MapDatasetOp's relationship with its associated function is as // follows: first M function params are dictated by the set // output shapes and types, the next N are the last Ninputs from MapDataset // op. The MapDataset op always has N+1 inputs. // TODO(jpienaar): Avoid this lookup. auto module = op->getParentOfType<ModuleOp>(); auto f = module.lookupSymbol<func::FuncOp>(op.getF());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// For this test case the first 12 inputs to the function f below will be // from ParallelMapDataset, while the remaining one is a captured input. %75 = "tf.MapDataset"(%74, %arg3) { device = "/job:localhost/replica:0/task:0/device:CPU:0", f = @__inference_Dataset_map_lambda_96150, f._tf_data_function = true,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)