- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ResourceGather (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
), (TF_AssignVariableOp $m_resource, $new_m, (CreateConstBoolAttrFalse)), (TF_AssignVariableOp $v_resource, $new_v, (CreateConstBoolAttrFalse)) ] >; // Pattern to decompose tf.ResourceGather into tf.ReadVariable and tf.GatherV2. def DecomposeResourceGather : Pat< (TF_ResourceGatherOp:$old_result $resource, $indices, $batch_dims, $validate_indices), (TF_GatherV2Op:$dest
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
// 2) Convert composite functions like lstm/rnns, along with proper function // inlining & dce. // 3) Lower static tensor list pass. // This decomposes resource ops like ResourceGather into read-variable op // followed by gather. This is used when the saved model import path is used // during which resources don't get frozen in the python layer. pass_manager->addNestedPass<mlir::func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0)