Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 28 for init_values (0.33 sec)

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

        inputs.push_back(std::move(tensor));
      }
    
      std::vector<tensorflow::TensorValue> input_values;
      for (tensorflow::Tensor& tensor : inputs) {
        input_values.emplace_back();
        input_values.back().tensor = &tensor;
      }
    
      tensorflow::OpKernelContext::Params params;
      params.inputs = input_values;
      params.device = runner->device();
      params.op_kernel = runner->op_kernel();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc

      void rewrite(AddOp op, PatternRewriter& rewriter) const override {
        DeferRhsTransposeForBinaryOp(op, rewriter);
      }
    };
    
    // Rewrites the `reduce_window(transpose(%activation), %init_value)` patterns to
    // `transpose(reduce_window(%activation), %init_value)`, deferring the transpose
    // to the result. The reduce function should be equivalent to
    // `stablehlo.maximum`, representing max pooling.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  3. tensorflow/cc/ops/while_loop_test.cc

      }
    
      template <typename T>
      void Run(const std::vector<Input::Initializer>& input_values,
               const std::vector<T>& expected_output_values) {
        ClientSession session(scope_);
    
        DCHECK_EQ(input_values.size(), inputs_.size());
        ClientSession::FeedType feeds;
        for (int i = 0; i < inputs_.size(); ++i) {
          feeds.emplace(inputs_[i], input_values[i]);
        }
    
        std::vector<Tensor> out_tensors;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 13 22:30:58 UTC 2023
    - 6.4K bytes
    - Viewed (0)
  4. tensorflow/c/while_loop_test.cc

      }
    
      void Run(std::initializer_list<int> input_values) {
        Run(outputs_, input_values);
      }
    
      void Run(const std::vector<TF_Output>& run_outputs,
               std::initializer_list<int> input_values) {
        DCHECK_EQ(inputs_.size(), input_values.size());
        std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs(inputs_.size());
        int i = 0;
        for (int v : input_values) {
          inputs[i] = {inputs_[i].oper, Int32Tensor(v)};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 11 06:05:56 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

     public:
      using ConvertReduceOpToTfOp::ConvertReduceOpToTfOp;
    
      LogicalResult MatchInitValue(Value init_value) const override {
        auto type = mlir::cast<ShapedType>(init_value.getType()).getElementType();
        if (mlir::isa<FloatType>(type)) {
          float const_value;
          if (failed(GetConstantSplatValue<float>(init_value, const_value)) ||
              const_value != 1.0)
            return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

        } else {
          llvm::SmallVector<Value> input_values(node_output_tensors.size(),
                                                scalar_one);
          for (const auto& [output_index, tensor_index, tensor_value] :
               node_output_tensors) {
            input_values[tensor_index] = tensor_value;
          }
          identity_op = builder.create<TF::IdentityNOp>(
              new_loc, TypeRange(ValueRange(input_values)), input_values);
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
  7. tensorflow/cc/training/queue_runner_test.cc

    constexpr char kVarOpName[] = "var";
    
    GraphDef BuildSimpleGraph() {
      Scope root = Scope::NewRootScope();
      auto init_value = Const(root, 0);
      auto var = Variable(root.WithOpName(kVarOpName), TensorShape({}),
                          DataType::DT_INT32);
      auto assign = Assign(root.WithOpName(kAssignOpName), var, init_value);
      auto count = CountUpTo(root.WithOpName(kCountUpToOpName), var, 10);
      Square(root.WithOpName(kSquareOpName), var);  // NOLINT
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Sep 21 06:27:51 UTC 2019
    - 14.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc

        return "Unfreeze large constants.";
      }
    
      void runOnOperation() override;
    
     private:
      Option<int64_t> CreateSizeThresholdInBytesOption(const int64_t init_value) {
        return Option<int64_t>(
            *this, "size_threshold_in_bytes", llvm::cl::init(init_value),
            llvm::cl::desc(
                "Lower threshold of the constant size for unfreezing. Constants "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 16 15:04:53 UTC 2023
    - 14K bytes
    - Viewed (0)
  9. platforms/core-configuration/model-core/src/integTest/groovy/org/gradle/api/provider/PropertyAssignmentIntegrationTest.groovy

        }
    
        def "test Groovy eager collection types assignment for #description"() {
            def initValue = inputType.contains("Map<") ? "[:]" : "[]"
            def inputDeclaration = "$inputType input = $initValue"
            groovyBuildFile(inputDeclaration, inputValue, operation)
    
            expect:
            runAndAssert("myTask", expectedResult)
    
            where:
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Thu Dec 28 14:39:49 UTC 2023
    - 36.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

    TEST(ResourceOperationSafetyAnalysisTest, Loop) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output init_value = ops::Placeholder(root.WithOpName("init"), DT_FLOAT);
      Output loop_cond = ops::Placeholder(root.WithOpName("init"), DT_BOOL);
      Output enter_value =
          ops::internal::Enter(root.WithOpName("enter"), init_value, "fr");
      ops::Merge iv(root.WithOpName("iv"), {enter_value, enter_value});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
Back to top