Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for add_input (0.14 sec)

  1. tensorflow/cc/tools/freeze_saved_model.cc

                                         NodeDef* identity_node) {
      identity_node->set_name(node.name());
      identity_node->set_op("Identity");
      (*identity_node->mutable_attr())["T"] = node.attr().at("dtype");
      identity_node->add_input(node.input(0));
    }
    
    // Returns the name of the VarHandleOp that provides input (possibly indirectly)
    // to node with node_name. A typical indirect chain of nodes (that can occur due
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 11 08:05:36 UTC 2023
    - 11.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      // as later when edges are added to the Node using Graph::AddEdge the
      // associated NodeDef is not updated.
      for (int i = 0, e = inst->getNumOperands(); i < e; ++i) {
        node_def->add_input();
      }
      if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) {
        node_def->set_device(std::string(attr.getValue()));
      }
    
      // Add the node debug info.
      TF_RETURN_IF_ERROR(ConvertLocation(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

      NodeDef node_def;
      tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op);
      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
      OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr());
    
      const tensorflow::OpRegistrationData* op_reg_data;
      status->status =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/ops/math_ops.cc

      AbstractOperationPtr op_ptr(ctx->CreateOperation());
      TF_RETURN_IF_ERROR(op_ptr->Reset("Mul", raw_device_name));
      TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name));
      TF_RETURN_IF_ERROR(op_ptr->AddInput(x));
      TF_RETURN_IF_ERROR(op_ptr->AddInput(y));
      int num_retvals = 1;
      return op_ptr->Execute(absl::MakeSpan(z, 1), &num_retvals);
    }
    
    // Op: Conj()
    // Summary: Returns the complex conjugate of a complex number.
    //
    // Description:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 10 19:11:36 UTC 2022
    - 12.2K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
                    ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInput(input));
      forward_op_->inputs.push_back(input);
      return absl::OkStatus();
    }
    Status AddInputList(AbstractOperation* op_,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_test_util.cc

                     TF_Status* s, const char* name, TF_Operation** op,
                     bool check) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "AddN", name);
      TF_Output add_inputs[2] = {{l, 0}, {r, 0}};
      TF_AddInputList(desc, add_inputs, 2);
      *op = TF_FinishOperation(desc, s);
      if (check) {
        ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        ASSERT_NE(*op, nullptr);
      }
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Oct 15 03:16:52 UTC 2021
    - 17.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir

      %bias = arith.constant dense<[7.11401462, 7.05456924]> : tensor<2xf32>
    
      %q_input= "quantfork.qcast"(%input) : (tensor<1x3x4x3xf32>) -> tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>>
      %dq_input= "quantfork.dcast"(%q_input) : (tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>>) -> tensor<1x3x4x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 19:32:28 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

      SmallVector<Value, 8> operands_;
      llvm::StringMap<Attribute> attrs_;
      std::unique_ptr<OperationState> state_;
      // This is the index of the next ODS operand that will be added with AddInput
      // or AddInput;
      int current_ods_input_ = 0;
      const tensorflow::OpDef* op_def_ = nullptr;
      const char* op_name_ = nullptr;
      string tf_op_type_;
      // TODO(srbs): Use this.
      string device_name_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  9. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      Status SetDeviceName(const char* name) override {
        // TODO(srbs): Implement this.
        device_name_ = name;
        return absl::OkStatus();
      }
    
      Status AddInput(AbstractTensorHandle* input) override {
        GraphTensor* t = dyn_cast<GraphTensor>(input);
        if (!t) {
          return tensorflow::errors::InvalidArgument(
              "Unable to cast input to GraphTensor");
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 20:00:09 UTC 2024
    - 15.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

        // Rewrite pad_op before Convolutioin.
        if (failed(HandlePad(pad_op, filter_shape[0], block_size))) return;
        auto pad_input = pad_op.getInput();
        if (auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(pad_input)) {
          // Change on device function type/shape.
          HandleFuncOp(block_arg.getOwner()->getParentOp());
        }
      }
    
      // Handle Conv2D input, stride and filter.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
Back to top