Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 199 for tensor_b (0.22 sec)

  1. tensorflow/compiler/jit/deadness_analysis_test.cc

    TEST(DeadnessAnalysisTest, Recv) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output recv_a = ops::_Recv(root.WithOpName("recv_a"), DT_FLOAT, "tensor_a",
                                 "sender", 0, "receiver");
      Output recv_b = ops::_Recv(root.WithOpName("recv_b"), DT_FLOAT, "tensor_b",
                                 "sender", 0, "receiver");
      Output add = ops::Add(root.WithOpName("add"), recv_a, recv_b);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

    TEST(XlaCompilationTest, ClusterControlTrigger) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      Output recv_a = ops::_Recv(root.WithOpName("recv_a"), DT_BOOL, "tensor_a",
                                 "sender", 0, "receiver");
      Output recv_b = ops::_Recv(root.WithOpName("recv_b"), DT_BOOL, "tensor_b",
                                 "sender", 0, "receiver");
      Output const_a = ops::Const(root.WithOpName("const_a"), 42);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
  3. tensorflow/c/eager/tape.h

        Gradient* grad_tensor = grad[grad_index];
        if (grad_tensor != nullptr) {
          int64_t tensor_id = vspace_.TensorId(grad_tensor);
          targets.push_back(tensor_id);
          if (sources_set.find(tensor_id) != sources_set.end()) {
            sources_that_are_targets.emplace(
                tensor_id, vspace_.TapeTensorFromGradient(grad_tensor));
          }
          Gradient* in_grad = in_grads[grad_index];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 12:40:29 UTC 2024
    - 47.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/deadness_analysis.cc

      // "tensor_id() is live and evaluates to true".
      //
      // If `must_be_true()` is false then this SymbolPredicate represents the
      // proposition "tensor_id() is live (and may evaluate to any value)"
      TensorId tensor_id() const { return tensor_id_; }
      bool must_be_true() const { return must_be_true_; }
    
     private:
      TensorId tensor_id_;
      bool must_be_true_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
  5. tensorflow/c/tf_tensor.cc

        auto* ret = emptyTensor->tensor;
        delete emptyTensor;
        return ret;
      }
    
      Tensor tensor;
      if (!tensor.CopyFrom(src, src.shape())) {
        return nullptr;
      }
      return new tensorflow::TensorInterface(std::move(tensor));
    }
    
    // Non-static for testing.
    TF_Tensor* TF_TensorFromTensor(const tensorflow::Tensor& src, Status* status) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 21:57:32 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      }];
    
      let arguments = (ins
        Arg<TF_Int32Tensor, [{int32 `Tensor`.}]>:$arr,
        Arg<TF_Int32Tensor, [{non-negative int32 scalar `Tensor`.}]>:$size,
        Arg<TensorOf<[TF_Float32, TF_Float64, TF_Int32, TF_Int64]>, [{is an int32, int64, float32, or float64 `Tensor` with the same
    shape as `arr`, or a length-0 `Tensor`, in which case it acts as all weights
    equal to 1.}]>:$weights
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  7. tensorflow/cc/framework/ops.h

      DataType data_type() const { return data_type_; }
      Status status() const { return status_; }
      const Tensor& tensor() const { return tensor_; }
    
     private:
      Status status_;
      Output output_ = Output(Operation(nullptr), 0);
      Tensor tensor_;
      const std::string node_name_ = "";
      int32 index_ = 0;
      DataType data_type_ = DT_INVALID;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

    //
    // Conditions for `tfl.fully_connected` conversion:
    //   * Input tensors are per-tensor uniform quantized (i8->f32)
    //     tensors.
    //   * The filter tensor is constant a per-tensor uniform quantized (i8->f32)
    //     tensor. The quantization dimension should be 1 (the non-contracting
    //     dimension).
    //   * Output tensors are per-tensor uniform quantized (i8->f32) or
    //     per-channel uniform quantized (i32->f32) tensors.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td

              (TF_SubOp $max_range, $min_range),
              (TF_ConstOp (GetF32Scalar<255>)))),
          $min_range),
        /*truncate=*/ConstBoolAttrFalse),
      [(TensorOf<[TF_Qint8, TF_Quint8]> $input),
       (TF_Float32Tensor $min_range),
       (TF_Float32Tensor $max_range),
       (TensorOf<[TF_Float32, TF_Bfloat16]> $result)]>;
    
    //===----------------------------------------------------------------------===//
    // Difference op patterns.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 24.7K bytes
    - Viewed (0)
  10. tensorflow/c/tf_tensor.h

    TF_CAPI_EXPORT extern TF_Tensor* TF_TensorMaybeMove(TF_Tensor* tensor);
    
    // Destroy a tensor.
    TF_CAPI_EXPORT extern void TF_DeleteTensor(TF_Tensor*);
    
    // Return the type of a tensor element.
    TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*);
    
    // Set a new shape for the Tensor.
    TF_CAPI_EXPORT extern void TF_SetShape(TF_Tensor* tensor, const int64_t* dims,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 06 16:40:30 UTC 2024
    - 6.3K bytes
    - Viewed (0)
Back to top