Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 38 for zeros_like (0.17 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    }
    
    //===----------------------------------------------------------------------===//
    // ZerosLike
    //===----------------------------------------------------------------------===//
    
    // -----
    
    // CHECK-LABEL: @zeros_like
    // CHECK-SAME:  (%[[ARG:.*]]: tensor<2x?xf32>)
    func.func @zeros_like(%arg0: tensor<2x?xf32>) -> tensor<2x?xf32> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  2. tensorflow/cc/gradients/math_grad.cc

      auto gathered = Gather(scope, params, zero_clipped_indices);
      // Replace gathered params of negative indices with 0.
      auto zero_slice = ZerosLike(scope, gathered);
      return SelectV2(scope, is_positive, gathered, zero_slice);
    }
    
    Status UnsortedSegmentMinOrMaxGrad(const Scope& scope, const Operation& op,
                                       const std::vector<Output>& grad_inputs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.h

    // gradients.
    //
    // `ZerosLike` is not expected to be called and returns a nullptr. The creation
    // of default zeros grads is handled by the `DefaultGradientFunction` registered
    // for each op.
    // TODO(srbs): We need to define `ZerosLike` here to keep the compiler happy.
    // Figure out a way to avoid this.
    // TODO(srbs): Should ZerosLike check-fail instead of returning nullptr?
    class TapeTensor {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Sep 26 10:27:05 UTC 2022
    - 6.9K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/ops/array_ops.cc

      TF_RETURN_IF_ERROR(op_ptr->AddInputList(input));
      int num_retvals = output.size();
      return op_ptr->Execute(output, &num_retvals);
    }
    
    // Op: ZerosLike()
    // Summary: Returns a tensor of zeros with the same shape and type as x.
    //
    // Description:
    Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x,
                     AbstractTensorHandle** y, const char* name,
                     const char* raw_device_name) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 10 19:11:36 UTC 2022
    - 6.7K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/ops/update_cpp_ops.sh

      --output_dir="${current_dir}" \
      --api_dirs="${api_dir}" \
      --source_dir=third_party/tensorflow"
    
    ${generate} \
      --category=array \
      Identity \
      IdentityN \
      ZerosLike \
      Shape \
      ExpandDims \
      OnesLike
    
    ${generate} \
      --category=math \
      Mul \
      Conj \
      AddV2 \
      MatMul \
      Neg \
      Sum \
      Sub \
      Div \
      DivNoNan \
      Exp \
      Sqrt \
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 17 17:54:34 UTC 2022
    - 1.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/attributes.mlir

      func.return
    }
    
    // CHECK-LABEL: func @variant
    func.func @variant(%arg: tensor<!tf_type.variant>) {
      // CHECK: tf.ZerosLike
      // CHECK-SAME: T = !corert.variant
      %0 = "tf.ZerosLike"(%arg) {device = "/device:CPU:0", T = !tf_type.variant} : (tensor<!tf_type.variant>) -> tensor<!tf_type.variant>
      func.return
    }
    
    // Checks that TF quantized attrs are lowered to the corert types
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/ops/array_ops.h

                     const char* name = nullptr,
                     const char* raw_device_name = nullptr);
    
    // Returns a tensor of zeros with the same shape and type as x.
    Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* const x,
                     AbstractTensorHandle** y, const char* name = nullptr,
                     const char* raw_device_name = nullptr);
    
    // Returns the shape of a tensor.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 10 19:11:36 UTC 2022
    - 2.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      }];
    
      let hasOptions = 1;
    }
    
    def TFL_ZerosLikeOp: TFL_Op<"zeros_like", [
        PredOpTrait<"input and output must have same element type",
          TFL_TCresVTEtIsSameAsOp<0, 0>>,
        SameOperandsAndResultShape,
        Pure]> {
      let summary = "ZerosLike operator";
    
      let description = [{
        Returns a tensor of zeros with the same shape and type as the input tensor.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  9. tensorflow/c/eager/gradients.cc

    int64_t ToId(const AbstractTensorHandle* t) {
      return static_cast<int64_t>(reinterpret_cast<uintptr_t>(t));
    }
    
    Status ZerosLike(AbstractContext* ctx, AbstractTensorHandle* t,
                     AbstractTensorHandle** result) {
      AbstractOperationPtr op(ctx->CreateOperation());
      TF_RETURN_IF_ERROR(op->Reset("ZerosLike", /*raw_device_name=*/nullptr));
      if (isa<tracing::TracingOperation>(op.get())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  10. src/runtime/iface_test.go

    		{name: "E32", fn: func() { e = zero32 }},
    		{name: "E64", fn: func() { e = zero64 }},
    		{name: "Estr", fn: func() { e = zerostr }},
    		{name: "Eslice", fn: func() { e = zeroslice }},
    		{name: "Econstflt", fn: func() { e = 99.0 }}, // constants do not allocate
    		{name: "Econststr", fn: func() { e = "change" }},
    		{name: "I8", fn: func() { i1 = eight8I }},
    		{name: "I16", fn: func() { i1 = zero16I }},
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Dec 13 18:45:54 UTC 2021
    - 7.5K bytes
    - Viewed (0)
Back to top