- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 205 for dims1 (0.06 sec)
-
tensorflow/c/eager/unified_api_testutil.h
// Return a tensor handle with given type, values and dimensions. template <class T, TF_DataType datatype> Status TestTensorHandleWithDims(AbstractContext* ctx, const T* data, const int64_t* dims, int num_dims, AbstractTensorHandle** tensor) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TFE_Context* eager_ctx =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, ForwardOperation*); Status SetAttrShape(AbstractOperation*, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation*); Status SetAttrFunction(AbstractOperation*, const char* attr_name, const AbstractOperation* value, ForwardOperation*);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:35 UTC 2021 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-tfl-stablehlo-broadcast.mlir
func.return %0 : tensor<1x2x2xi32> } } // CHECK: module { // CHECK-NEXT: func @main(%arg0: tensor<1x2xi32>) -> tensor<1x2x2xi32> { // CHECK-NEXT: %0 = stablehlo.broadcast_in_dim %arg0, dims = [1, 2] : (tensor<1x2xi32>) -> tensor<1x2x2xi32> // CHECK-NEXT: return %0 : tensor<1x2x2xi32> // CHECK-NEXT: }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 16 05:09:09 UTC 2022 - 704 bytes - Viewed (0) -
tensorflow/cc/experimental/base/public/tensor.h
Tensor& operator=(Tensor&&) = default; // Returns the number of dimensions in the tensor. Can be -1, which represents // unknown rank. int dims() const; // Returns the number of elements in dimension `d`. // REQUIRES: `0 <= d < dims()` int64_t dim_size(int d) const; // Returns a pointer to the underlying data buffer. void* data() const; // Returns the data type of the tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 28 20:10:33 UTC 2020 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
// Checks if the value has static shape. def HasStaticShapeConstraint : Constraint<CPred<"HasStaticShape($0)">>; // Checks if the value has static shape at given dims. class HasStaticShapeAtDimsConstraint<string dims> : Constraint< CPred<"HasStaticShapeAtDims($0, {"# dims #"})">>; // The rewrite rule cannot replace a value with itself, so we work around // by cloning the root op to replicate that value. The old op will get folded.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/pre_calibration_component.mlir
// [b, 0, 1, f]). The weight constant is folded into [0, 1, i, o] format. // CHECK-DAG: %[[CST:.+]] = stablehlo.constant dense<3.000000e+00> : tensor<3x3x8x8xf32> // CHECK: %[[TRANSPOSE_1:.+]] = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x8x4x4xf32>) -> tensor<1x4x4x8xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
TEST(BitcastOpTest, TestImpossibleCast) { Tensor int8_input(DT_UINT8, {1}); TestBitcastOp(&int8_input, DT_UINT32, TensorShape(), error::INVALID_ARGUMENT); } PartialTensorShape S(std::initializer_list<int64_t> dims) { return PartialTensorShape(dims); } TEST(BitcastOpTest, TestShapeInference_LargerShape) { const OpRegistrationData* reg; TF_CHECK_OK(OpRegistry::Global()->LookUp("Bitcast", ®)); OpDef op_def = reg->op_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.h
class Operation; namespace TF { class AddV2Op; class SubOp; class MulOp; class DivOp; class RealDivOp; // Verifies an reduction op's `input` and reduction `dims`. LogicalResult VerifyReductionInputAndDims(Value input, Value dims, Location loc); // A type range with description (in singular form) attached to it. using TypeRangeWithDesc = std::pair<TypeRange, StringRef>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor; dlm_tensor->manager_ctx = tf_dlm_tensor_ctx; dlm_tensor->deleter = &DLManagedTensorDeleter; dlm_tensor->dl_tensor.device = tf_dlm_context; int ndim = tensor->dims(); dlm_tensor->dl_tensor.ndim = ndim; dlm_tensor->dl_tensor.data = tf_dlm_data; dlm_tensor->dl_tensor.dtype = tf_dlm_type; std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 12.8K bytes - Viewed (0)