Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 10 for dims1 (0.04 sec)

  1. tensorflow/c/c_api_test.cc

      const char tensor1[] = {5, 7};
      const int64_t dims1[] = {1, 2};
      const size_t ndims1 = TF_ARRAYSIZE(dims1);
    
      const char tensor2[] = {2, 4, 6, 8};
      const int64_t dims2[] = {2, 2};
      const size_t ndims2 = TF_ARRAYSIZE(dims2);
    
      auto desc = init("list(tensor)");
      TF_Tensor* tmp[] = {
          Int8Tensor(dims1, ndims1, tensor1),
          Int8Tensor(dims2, ndims2, tensor2),
      };
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions.mlir

      %5 = stablehlo.dynamic_broadcast_in_dim %1, %4, dims = [3] : (tensor<16xf32>, tensor<4xindex>) -> tensor<?x28x28x16xf32>
      %6 = stablehlo.add %3, %5 : tensor<?x28x28x16xf32>
      %7 = shape.shape_of %6 : tensor<?x28x28x16xf32> -> tensor<4xindex>
      %8 = stablehlo.dynamic_broadcast_in_dim %2, %7, dims = [] : (tensor<f32>, tensor<4xindex>) -> tensor<?x28x28x16xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 49.8K bytes
    - Viewed (0)
  3. tensorflow/c/kernels_test.cc

    #else
      const char* device_name_ = tensorflow::DEVICE_CPU;
    #endif
    };
    
    // Validates that the tensor has shape and type corresponding to
    // dims and dtype.
    void validate_tensor(TF_Tensor* tensor, int64_t* dims, int64_t num_dims,
                         TF_DataType dtype);
    
    // Copies data of length tensor_size_bytes from values to tensor.
    template <typename T>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.h

    // Returns the shape of the Tensor referenced by `output` in `graph`
    // into `dims`. `dims` must be an array large enough to hold `num_dims`
    // entries (e.g., the return value of TF_GraphGetTensorNumDims).
    //
    // If the number of dimensions in the shape is unknown or the shape is
    // a scalar, `dims` will remain untouched. Otherwise, each element of
    // `dims` will be set corresponding to the size of the dimension. An
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir

          %2 = stablehlo.reshape %1 : (tensor<i32>) -> tensor<1xi32>
          %3 = stablehlo.concatenate %2, %cst_0, dim = 0 : (tensor<1xi32>, tensor<1xi32>) -> tensor<2xi32>
          %4 = stablehlo.dynamic_broadcast_in_dim %arg2, %3, dims = [1] : (tensor<3xf32>, tensor<2xi32>) -> tensor<?x3xf32>
          %5 = stablehlo.add %0, %4 : tensor<?x3xf32>
          return %5 : tensor<?x3xf32>
        }
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 91.6K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_test.cc

      TFE_DeleteContextOptions(opts);
    
      std::vector<int64_t> dims(4, 1);
      TFE_Op* op = TFE_NewOp(ctx, "AvgPool", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TF_Tensor* tensor =
          TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float));
      float tensor_data[] = {1};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  7. tensorflow/cc/gradients/math_grad.cc

      auto gx = SelectV2(scope, c, grad_inputs[0], zeros);
      auto x_shape = Shape(scope, x);
      auto output_shape = Shape(scope, op.output(0));
    
      // Reduce away broadcasted leading dims.
      auto reduce_x = internal::BroadcastGradientArgs(scope, x_shape, output_shape);
      auto gx_sum =
          ReduceSum(scope, gx, /*axis=*/reduce_x.r0, ReduceSum::KeepDims(true));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  8. CHANGELOG/CHANGELOG-1.31.md

    - Scheduler framework: PreBind implementations are now allowed to return Pending and Unschedulable status...
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Jun 12 20:34:14 UTC 2024
    - 60.3K bytes
    - Viewed (0)
  9. CHANGELOG/CHANGELOG-1.3.md

    * Increase request timeout based on termination grace period ([#31275](https://github.com/kubernetes/kubernetes/pull/31275), [@dims](https://github.com/dims))
    * Skip safe to detach check if node API object no longer exists ([#30737](https://github.com/kubernetes/kubernetes/pull/30737), [@saad-ali](https://github.com/saad-ali))
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Thu Dec 24 02:28:26 UTC 2020
    - 84K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        // Replace the original strided_slice.
        uint64_t revised_begin_mask = strided_slice_op.getBeginMask();
        uint64_t revised_end_mask = strided_slice_op.getEndMask();
        // Since we expand the dims, we need to apply them to the begin_mask &
        // end_mask.
        revised_begin_mask |= strided_slice_op.getNewAxisMask();
        revised_end_mask |= strided_slice_op.getNewAxisMask();
    
        // Enforce operator precedence.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
Back to top