- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 213 for se_shape (0.24 sec)
-
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
} }]; let hasCanonicalizer = 1; } def TFR_GetShapeOp : TFR_Op<"get_shape", [Pure]> { let description = [{ The `get_shape` operation gets the shape of a tfr.tensor and returns !shape.shape type. Example: ```mlir %1 = "tfr.get_shape"(%0) : !tfr.tensor -> !shape.shape %1 = tfr.get_shape %0 -> !shape.shape ``` }]; let arguments = (ins TFR_TensorType:$arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.h
// Represents a (partially-defined) shape. typedef struct TF_Shape { int num_dims; // Must be >= -1; -1 represents unknown rank. int64_t* dim_sizes; } TF_Shape; // Add a new parameter to a TensorFlow Function. TF_AbstractTensor* TF_AddFunctionParameter(TF_ExecutionContext* func, TF_DataType dtype, TF_Shape shape, TF_Status* s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Oct 24 11:27:00 UTC 2021 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
TorchAvgPoolData data; auto op_type = mlir::cast<RankedTensorType>(op.getOperand(0).getType()); data.n = op_type.getShape()[0]; data.c = op_type.getShape()[1]; data.h_in = op_type.getShape()[2]; data.w_in = op_type.getShape()[3]; std::vector<int32_t> kernel_size; GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "kernel_size",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/c/tf_tensor_internal.h
size_t ByteSize() const override; void* Data() const override; bool IsAligned() const override; bool CanMove() const override; std::string SummarizeValue() const override; void SetShape(const int64_t* dims, int num_dims); Status ToTensor(tensorflow::Tensor* dst) const; Status BitcastFrom(const TensorInterface& from, DataType type, const int64_t* new_dims, int num_new_dims);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 24 20:38:55 UTC 2023 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/prepare_quantize_ptq.mlir
// CHECK-SAME: quant.uniform<i8:f32, 0.010039215461880554:-1> // CHECK: %[[dq1:.*]] = "quantfork.dcast"(%[[q1]]) // CHECK-SAME: quant.uniform<i8:f32, 0.010039215461880554:-1> // CHECK: %[[reshape:.*]] = "tf.Reshape"(%[[dq1]] // CHECK: %[[q2:.*]] = "quantfork.qcast"(%[[reshape]]) // CHECK-SAME: quant.uniform<i8:f32, 0.010039215461880554:-1> // CHECK: %[[dq2:.*]] = "quantfork.dcast"(%[[q2]]) // CHECK-SAME: quant.uniform<i8:f32, 0.010039215461880554:-1>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 10:21:29 UTC 2023 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
} auto values_type = RankedTensorType::get( result_type.getShape(), mlir::cast<quant::UniformQuantizedType>(result_type.getElementType()) .getStorageType()); DenseElementsAttr reshaped_elements = dense_elements.reshape(values_type); rewriter.replaceOpWithNewOp<QConstOp>(op, TypeAttr::get(result_type),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/README.md
%cst_0 = arith.constant dense<2> : tensor<1xi32> %cst_1 = arith.constant dense<[2, 1]> : tensor<2xi32> %0 = "tfl.reshape"(%arg0, %cst) {tac.device = "GPU", tac.inference_type = "FLOAT"} : (tensor<1xf32>, tensor<4xi32>) -> tensor<1x1x1x1xf32> %1 = "tfl.reshape"(%arg1, %cst) {tac.device = "GPU", tac.inference_type = "FLOAT"} : (tensor<1xf32>, tensor<4xi32>) -> tensor<1x1x1x1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 29 18:32:13 UTC 2022 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
Value temp_padding = CreateConstValue<int32_t>(builder, loc, {num_dims, 2}, padding_values); SmallVector<int64_t> output_shape(input_shape.getShape().begin(), input_shape.getShape().end()); for (int i : spatial_dims) { output_shape[i] += padding_values[2 * i] + padding_values[2 * i + 1]; } return builder.create<TF::PadV2Op>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
// Reshapes the R1 length to a scalar. auto reshape = builder.create<TF::ReshapeOp>( length.getLoc(), ArrayRef<Type>{RankedTensorType::get( {}, getElementTypeOrSelf(current_size.getType()))}, ArrayRef<Value>{current_size, cutil::GetR1Const({}, builder, length.getLoc())}); length.getLength().replaceAllUsesWith(reshape); } length.erase(); return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.h
// ShapeContainerT is any type with the following methods: // bool hasRank() // ArrayRef<int64_t> getShape() // This includes mlir::TF::ShapeAttr and mlir::ShapedType. template <typename ShapeContainerT> void SetTensorShapeProto(ShapeContainerT shape, TensorShapeProto* proto) { if (shape.hasRank()) { for (int64_t dim : shape.getShape()) { proto->add_dim()->set_size(mlir::ShapedType::isDynamic(dim) ? -1 : dim); } } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.9K bytes - Viewed (0)