Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetFuncAttr (0.32 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc

          has_resources = true;
        }
      }
    
      if (!in_order) {
        // Functions do not get reused in practice, so skip the check for if the
        // callee has been updated.
        StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr();
        MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym));
      }
      builder.setInsertionPoint(cluster_func_op);
      auto xla_launch_op = builder.create<TF::XlaLaunchOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc

            }
          }
        }
    
        cluster_func_op->setAttr(TF::kDynamicArgIndexAttr,
                                 builder.getI32ArrayAttr(dynamic_shape_arg_index));
    
        FlatSymbolRefAttr func_attr = cluster_func_op.getFuncAttr();
        func::FuncOp func =
            cluster_func_op->getParentOfType<ModuleOp>().lookupSymbol<func::FuncOp>(
                func_attr.getValue());
    
        // Update the marked argument with dynamic shapes.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top