Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 60 of 1,512 for q_builder (0.18 sec)

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

      push.replaceAllUsesWith(push.getElem());
      OpBuilder builder(push);
      // Read the current buffer and size.
      auto stack_val =
          cutil::ReadLocalVariable(push.getHandle(), builder, push.getLoc());
      auto index =
          cutil::ReadLocalVariable(it->getSecond(), builder, push.getLoc());
      stack_val = cutil::SetElement(index, stack_val, push.getElem(), builder,
                                    push.getLoc());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc

          }
        }
        return WalkResult::advance();
      });
      OpBuilder builder(replicate);
      OpBuilder inner_builder = OpBuilder::atBlockBegin(&cluster.getBody().front());
      // mapping from original operand to new XlaAllReduce op.
      llvm::DenseMap<Value, Value> orig_to_new;
    
      for (Value bcast : bcasts) {
        if (failed(MoveBroadcastToCluster(builder, inner_builder, cluster,
    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/lite/experimental/tac/transforms/get_alternative_subgraph.cc

        OpBuilder* builder) {
      func::FuncOp cloned_func = func.clone();
      cloned_func.setPrivate();
      auto interface_name = GetInterFaceName(func);
      if (!interface_name.has_value()) {
        func.emitError("the func op does not have interface_name");
        return nullptr;
      }
    
      cloned_func->setAttr(
          kDevice, builder->getStringAttr(target_device_inference_type.hardware));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 12.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      }
    }
    
    // A helper class to inline TF::StatefulPartitionedCall ops
    struct Inliner : public InlinerInterface {
      Inliner(OpBuilder& builder, SymbolTable& symbol_table)
          : InlinerInterface(builder.getContext()),
            builder(builder),
            symbol_table(symbol_table) {}
    
      bool isLegalToInline(Operation* call, Operation* callable,
                           bool wouldBeCloned) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

    }
    
    Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) {
      op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true));
      return op;
    }
    
    // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal
    // is present, a tf._XlaSendFromHostV2 op is created instead.
    Operation* CreateSendFromHostOp(OpBuilder& builder, Location loc,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  6. src/internal/types/testdata/fixedbugs/issue45550.go

    // Use of this source code is governed by a BSD-style
    // license that can be found in the LICENSE file.
    
    package p
    
    type Builder /* ERROR "invalid recursive type" */ [T interface{ struct{ Builder[T] } }] struct{}
    type myBuilder struct {
    	Builder[myBuilder]
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Jan 17 19:54:25 UTC 2023
    - 314 bytes
    - Viewed (0)
  7. tensorflow/c/experimental/filesystem/plugins/windows/BUILD

        tags = [
            "manual",
            "nobuilder",
            "notap",
        ],
        visibility = ["//visibility:public"],
        deps = [":windows_filesystem_impl"],
    )
    
    # The real implementation of the filesystem.
    cc_library(
        name = "windows_filesystem_impl",
        srcs = ["windows_filesystem.cc"],
        copts = get_win_copts(),
        tags = [
            "manual",
            "nobuilder",
            "notap",
        ],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 20 06:38:26 UTC 2024
    - 936 bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc

      void RaiseTargetSubgraphsForBlock(
          Block& block, OpBuilder& builder, ModuleOp module, bool skip_cpu,
          int& func_count, const TF::SideEffectAnalysis::Info& side_effect_info);
    };
    
    // After raising ops and adding the Func & Call op, call this function
    // to set attributes specific to this pass.
    void AddAttrs(OpsAdded& ops_added, OpBuilder& builder, int func_count) {
      func::FuncOp& added_func_op = ops_added.func_op;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td

      CPred<"$_self.isa<DenseElementsAttr>()">,
      "non-opaque constant tensor">;
    
    def CreateGatherNdOp : NativeCodeCall<
        "$_builder.create<TF::GatherNdOp>($0.getLoc(), $0.getType(), $1, $2)">;
    
    def CreateTFCastOpI32 : NativeCodeCall<
        "CreateTFCastOpI32(&$_builder, $_loc, $0, $1)">;
    
    // Convert to std constant for statically shaped, non-opaque constants.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  10. tensorflow/c/ops.cc

                                 TF_Status* status) {
      auto* cc_builder = reinterpret_cast<OpDefBuilder*>(builder);
      TF_SetStatus(status, TF_OK, "");
      ::tensorflow::OpRegistry::Global()->Register(
          [cc_builder](::tensorflow::OpRegistrationData* op_reg_data) -> Status {
            Status result = cc_builder->Finalize(op_reg_data);
            delete cc_builder;
            return result;
          });
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 28 22:41:35 UTC 2022
    - 10.9K bytes
    - Viewed (0)
Back to top