- Sort Score
- Result 10 results
- Languages All
Results 81 - 86 of 86 for StringAttr (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
mlir::cast<BlockArgument>(entry.getSecond()).getArgNumber(); } if (lowered_callee != callee) { // Add the clone with a new name. lowered_callee.setName(StringAttr::get( callee->getContext(), llvm::formatv("{0}_stack_decomposed", callee.getName()).str())); SymbolTable(module).insert(lowered_callee); callee = lowered_callee; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
template <typename UniformQuantizedConvolutionOp> FailureOr<ElementsAttr> ConvertPaddingAttr( UniformQuantizedConvolutionOp op, const xla::ConvolutionDimensionNumbers &dnums, PatternRewriter &rewriter) { StringAttr conv_padding = op.getPaddingAttr(); SmallVector<int64_t> padding_nums; ShapedType lhs_shape = mlir::cast<ShapedType>(op.getLhs().getType()); ShapedType rhs_shape = mlir::cast<ShapedType>(op.getRhs().getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
bool IsSupportedHostInputOp(Operation* op) { TF::IteratorGetNextOp iter = llvm::dyn_cast<TF::IteratorGetNextOp>(op); if (!iter) return false; auto device = op->getAttrOfType<StringAttr>(kDeviceAttr); if (!device) return false; tensorflow::DeviceNameUtils::ParsedName parsed_device; if (!tensorflow::DeviceNameUtils::ParseFullName(device.getValue().str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
tensor_cast.getResult()); return success(); } LogicalResult ConvertLSTMCellSimpleToFusedLSTM::InitializeFromFuncAttributes() { auto attr = fused_func_op_->getAttrOfType<StringAttr>(kTFImplements); if (!attr) { return fused_func_op_.emitError() << "Invalid function attribute, expected " << kTFImplements << " attribute " "not found"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
#include "tsl/platform/status.h" namespace { using ::absl::StatusOr; using ::tensorflow::Status; using ::tensorflow::errors::InvalidArgument; StatusOr<mlir::StringAttr> GetPaddingAttr(TfLitePadding pad_params, mlir::Builder builder, mlir::Location loc) { auto padding = tflite::Padding::Padding_VALID;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
// indicating what `Method` has been applied to the quantized unit. call_op->setAttr( kQuantizationMethodAttr, xla_call_module_op->getAttrOfType<StringAttr>(kQuantizationMethodAttr)); rewriter.replaceOp(xla_call_module_op, call_op); } // Replaces a quantized `xla_call_module_op` with a `func::CallOp`. The callee
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0)