Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 205 for dims1 (0.05 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/tensorflow/ir/tf_traits.h

        return a;
      }
      int64_t rank = a.getRank();
      SmallVector<int64_t, 4> dims;
      dims.resize(rank);
      for (int i = 0, e = rank; i != e; i++) {
        int64_t dim0 = a.getDimSize(i);
        int64_t dim1 = b.getDimSize(i);
        dims[i] = (dim0 == ShapedType::kDynamic) ? dim1 : dim0;
      }
      return RankedTensorType::get(dims, a.getElementType());
    }
    }  // namespace detail
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.7K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test_util.cc

      constexpr int64_t dims[] = {100, 100};
      constexpr int num_elements = dims[0] * dims[1];
      float data[num_elements];
      for (int i = 0; i < num_elements; ++i) {
        data[i] = 1.0f;
      }
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0],
                                            sizeof(dims) / sizeof(int64_t), status);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/fold_constant_transpose.mlir

      %1 = stablehlo.transpose %0, dims = [1, 0] : (tensor<2x3xi32>) -> tensor<3x2xi32>
      return %1 : tensor<3x2xi32>
    }
    // CHECK: transpose
    
    // -----
    
    // Tests that transposing an argument cannot be folded.
    
    // CHECK-LABEL: transpose_arg
    func.func @transpose_arg(%arg0: tensor<2x3xf32>) -> tensor<3x2xf32> {
      %0 = stablehlo.transpose %arg0, dims = [1, 0] : (tensor<2x3xf32>) -> tensor<3x2xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 08:06:02 UTC 2024
    - 2.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/aot/codegen_test_h.golden

      }
      float& arg0(size_t dim0, size_t dim1) {
        return (*static_cast<float(*)[1][2]>(
            arg_data(0)))[dim0][dim1];
      }
      const float* arg0_data() const {
        return static_cast<const float*>(arg_data(0));
      }
      const float& arg0(size_t dim0, size_t dim1) const {
        return (*static_cast<const float(*)[1][2]>(
            arg_data(0)))[dim0][dim1];
      }
      int arg0_size() const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 01:20:01 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  6. OWNERS_ALIASES

        - mikedanese
      sig-auth-authorizers-reviewers:
        - deads2k
        - dims
        - enj
        - liggitt
        - mikedanese
        - ncdc
        - smarterclayton
        - sttts
        - thockin
        - wojtek-t
      sig-auth-certificates-approvers:
        - liggitt
        - mikedanese
        - smarterclayton
      sig-auth-certificates-reviewers:
        - deads2k
        - dims
        - enj
        - liggitt
        - mikedanese
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon May 20 23:08:03 UTC 2024
    - 11.3K bytes
    - Viewed (0)
  7. tensorflow/c/tf_tensor.cc

      static char empty;
      int64_t nelems = 1;
      std::vector<int64_t> dims;
      auto shape_dims = shape.dims();
      dims.reserve(shape_dims);
      for (int i = 0; i < shape_dims; ++i) {
        dims.push_back(shape.dim_size(i));
        nelems *= shape.dim_size(i);
      }
      CHECK_EQ(nelems, 0);
      return TF_NewTensor(
          dtype, reinterpret_cast<const int64_t*>(dims.data()), shape.dims(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 21:57:32 UTC 2024
    - 11.5K bytes
    - Viewed (0)
  8. cmd/gotemplate/OWNERS

    # See the OWNERS docs at https://go.k8s.io/owners
    
    reviewers:
      - bentheelder
      - dims
      - pohly
    approvers:
      - dims
      - pohly
    labels:
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Tue Aug 22 18:39:23 UTC 2023
    - 151 bytes
    - Viewed (0)
  9. cmd/prune-junit-xml/OWNERS

    # See the OWNERS docs at https://go.k8s.io/owners
    
    reviewers:
      - aojea
      - bentheelder
      - dims
      - liggitt
      - pohly
    approvers:
      - dims
      - liggitt
    labels:
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon May 22 18:01:38 UTC 2023
    - 175 bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/defer_activation_transpose.mlir

      %1 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x3x3x4xf32>) -> tensor<1x4x3x3xf32>
      %2 = stablehlo.add %1, %0 : tensor<1x4x3x3xf32>
      return %2 : tensor<1x4x3x3xf32>
    }
    // CHECK-SAME: (%[[ARG_0:.+]]: tensor<1x3x3x4xf32>) -> tensor<1x4x3x3xf32>
    // CHECK-DAG: %[[CONST_0:.+]] = stablehlo.constant
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 18 20:32:46 UTC 2024
    - 14.6K bytes
    - Viewed (0)
Back to top