Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 47 for add_input (0.15 sec)

  1. tensorflow/c/kernels/summary_op_test.cc

      def.set_device(DEVICE_CPU);
    
      AttrValue valuesTypeAttr;
      SetAttrValue(values->dtype(), &valuesTypeAttr);
      (*def.mutable_attr())["T"] = valuesTypeAttr;
    
      def.add_input(strings::StrCat("input1: ", DataTypeString(tags->dtype())));
      def.add_input(strings::StrCat("input2: ", DataTypeString(values->dtype())));
    
      std::unique_ptr<OpKernel> kernel =
          CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 6.7K bytes
    - Viewed (0)
  2. tensorflow/c/kernels_test.cc

      NodeDef def;
      def.set_op(op_name);
      def.set_name(node_name);
      def.set_device(device_name);
      def.add_input("input1");
      def.add_input("input2");
      def.add_input("input3");
      def.add_input("input3");
      def.add_input("input3");
    
      AttrValue v0;
      v0.set_type(DataType::DT_INT32);
      v0.set_i(3);
      (*def.mutable_attr())["NumInput3"] = v0;
      AttrValue v1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  3. tensorflow/c/kernels/bitcast_op_test.cc

      SetAttrValue(input_tensor->dtype(), &typeAttr);
    
      AttrValue outTypeAttr;
      SetAttrValue(out_type, &outTypeAttr);
    
      (*def.mutable_attr())["T"] = typeAttr;
      (*def.mutable_attr())["type"] = outTypeAttr;
    
      def.add_input(
          strings::StrCat("input1: ", DataTypeString(input_tensor->dtype())));
    
      std::unique_ptr<OpKernel> kernel =
          CreateOpKernel(DeviceType(DEVICE_CPU), nullptr, nullptr, def, 1, &status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jul 18 15:10:51 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/tests/device_compiler_test_helper.cc

            const std::pair<std::string, FunctionDefHelper::AttrValueWrapper>>
            attrs) {
      NodeDef node;
      node.set_name(std::string(name));
      node.set_op(std::string(op));
      for (const auto& input : inputs) node.add_input(input);
      for (const auto& attr : attrs)
        node.mutable_attr()->insert({attr.first, attr.second.proto});
      return node;
    }
    
    }  // namespace
    
    GraphDef DeviceCompilerSerializeTest::GetTestGraph(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

        Node* outside_compilation_node =
            lifted_arg_nodes_and_outside_compilation_nodes[i - original_arg_count]
                .second;
        node_def.add_input(absl::StrCat(outside_compilation_node->name(), ":", 0));
      }
      TF_ASSIGN_OR_RETURN(n, ReplaceNode(g, n, node_def));
    
      // Add edges from outside compilation nodes to call node.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  9. 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)
  10. tensorflow/c/experimental/ops/gen/cpp/golden/testing_ops.cc.golden

      TF_RETURN_IF_ERROR(op_ptr->Reset("AccumulatorApplyGradient", raw_device_name));
      TF_RETURN_IF_ERROR(MaybeSetOpName(op_ptr.get(), name));
      TF_RETURN_IF_ERROR(op_ptr->AddInput(handle));
      TF_RETURN_IF_ERROR(op_ptr->AddInput(local_step));
      TF_RETURN_IF_ERROR(op_ptr->AddInput(gradient));
      int num_retvals = 0;
      std::vector<AbstractTensorHandle*> dummy_outputs;
      return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 16 19:04:03 UTC 2023
    - 6.5K bytes
    - Viewed (0)
Back to top