- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getI1Type (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.cc
} else { return UnrankedTensorType::get(builder->getI1Type()); } } auto ranked_type = mlir::dyn_cast<RankedTensorType>(result_type); if (!ranked_type) return UnrankedTensorType::get(builder->getI1Type()); return RankedTensorType::get(ranked_type.getShape(), builder->getI1Type()); } Type InferReductionOpType(Value input, Value reduction_indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.cc
return builder.getIntegerType(64); case tflite::TensorType_STRING: return mlir::TF::StringType::get(builder.getContext()); case tflite::TensorType_BOOL: return builder.getI1Type(); case tflite::TensorType_INT16: return builder.getIntegerType(16); case tflite::TensorType_COMPLEX64: return mlir::ComplexType::get(builder.getF32Type()); case tflite::TensorType_COMPLEX128:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc
mlir::Value reduction_indices, bool keep_dims = false) -> mlir::Operation* { return opb .create<mlir::TF::AnyOp>(loc, opb.getI1Type(), input, reduction_indices, keep_dims) .getOperation(); }); // mlir::TF::ConstOp py::class_<mlir::TF::ConstOp>(m, "Tf_ConstOp")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K bytes - Viewed (0)