Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 103 for SetAttr (0.17 sec)

  1. tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc

                              default_value);
    }
    
    string AttrView::SetterMethod() const {
      if (!attr_.is_list()) {
        return absl::StrCat("SetAttr", toUpperCamel(attr_.full_type()));
      } else {
        return absl::StrCat("SetAttr", toUpperCamel(attr_.base_type()), "List");
      }
    }
    
    std::vector<string> AttrView::SetterArgs() const {
      if (attr_.full_type() == "string") {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc

      ConstOp zero = builder.create<ConstOp>(loc, zero_attr);
      zero->setAttr(kICIWeightDistributionMlirBridgeMarker,
                    builder.getBoolAttr(true));
      ConstOp shape = builder.create<ConstOp>(loc, shape_attr);
      shape->setAttr(kICIWeightDistributionMlirBridgeMarker,
                     builder.getBoolAttr(true));
      FillOp fill = builder.create<FillOp>(loc, shape, zero);
      fill->setAttr(kICIWeightDistributionMlirBridgeMarker,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 18:52:07 UTC 2024
    - 13.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc

      // Both the device attribute and compile_device_type attribute should be
      // propagated to the reduce function call.
      reduce_call->setAttr(kDeviceAttr,
                           reduce_dataset->getAttrOfType<StringAttr>(kDeviceAttr));
      reduce_call->setAttr(
          TF::kCompileDeviceTypeAttr,
          reduce_dataset->getAttrOfType<StringAttr>(TF::kCompileDeviceTypeAttr));
    
      SmallVector<Value, 4> if_returns;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

        if (auto device = res->getAttrOfType<StringAttr>("device")) {
          if (!device.getValue().empty())
            result_id->setAttr("device", device);
          else
            result_id->setAttr("device", compile_device_op);
        } else if (compile_device_op) {
          result_id->setAttr("device", compile_device_op);
        }
        res.getOutput().replaceAllUsesWith(compile_op->getResult(0));
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc

      auto func = builder.create<func::FuncOp>(module->getLoc(),
                                               kSessionInitFuncName, func_type);
      func->setAttr(kTfSavedModelExportedNamesAttr,
                    builder.getStrArrayAttr({kSessionInitFuncName}));
      func->setAttr(kTfSavedModelInitializerTypeAttr,
                    builder.getStringAttr(kTfSavedModelInitializerRestoreType));
      func.setVisibility(mlir::func::FuncOp::Visibility::Public);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc

      builder.setInsertionPoint(execute_launch);
      auto copy_with_layout = BuildCopyWithLayout(execute_launch, compile_launch,
                                                  get_layout, input, &builder);
      copy_with_layout->setAttr(kDeviceAttr, execute_launch.getDeviceAttr());
      execute.setOperand(execute_arg_index, copy_with_layout);
    }
    
    // Performs transformation for replicated inputs. Returns true if this is a
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

                                live_ins);
    
      func::FuncOp outlined_func =
          BuildFunction(live_ins.getArrayRef(), cluster_op, symbol_table, builder);
      cluster_op->setAttr(
          builder->getStringAttr(kFuncAttr),
          mlir::SymbolRefAttr::get(builder->getContext(), outlined_func.getName()));
    
      builder->setInsertionPoint(cluster_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc

                                FunctionFilter::FunctionFilterType type,
                                OpBuilder& builder) {
      for (Operation& op : func.front()) {
        if (type == FunctionFilter::SKIP_TARGET_ANNOTATION) {
          op.setAttr(kSkipTargetAnnotation, builder.getUnitAttr());
        } else if (type == FunctionFilter::INCLUDE_TARGET_ANNOTATION) {
          op.removeAttr(kSkipTargetAnnotation);
        }
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

      return ConstBytesAttr::get(builder->getContext(),
                                 StringRef(content.data(), content.size()));
    }
    
    }  // namespace
    
    void ConvertNMSPaddedFunc::RewriteFunc() {
      func_->setAttr(kTFImplements,
                     StringAttr::get(func_.getContext(), kTfNMSPadded));
      Value boxes = func_.getArgument(0);
      Value scores = func_.getArgument(1);
      Value max_output_size = func_.getArgument(2);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

        }
        results->push_back(value.getInt());
      }
      return success();
    }
    
    }  // namespace
    
    LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() {
      func_.eraseBody();
      func_.addEntryBlock();
      func_->setAttr(kTFImplements,
                     StringAttr::get(func_.getContext(), kMaxUnpooling));
    
      OpBuilder builder(func_.getBody());
      std::string custom_option_buffer;
      if (failed(CreateCustomOptions(custom_option_buffer))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
Back to top