- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 32 for type_attr (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/type_attr.mlir
Adrian Kuegel <******@****.***> 1648469008 -0700
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/type_attr.mlir
Tres Popp <******@****.***> 1648106860 -0700
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 07:31:32 UTC 2022 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.h
// Returns element type from attribute Type 'type_attr'. mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr); // Returns true if 'val' is not from Quantize op or // from Quantize Op with same quant type as 'qtype_attr' bool NotFromQuantOpOrSameQuantType(mlir::Value val, mlir::TypeAttr qtype_attr); } // namespace tflite
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/model/op_spec.cc
for (const OpDef::ArgDef& arg_def : op_def.input_arg()) { ArgSpec arg = ArgSpec::CreateInput(arg_def, input_args_.size()); input_args_.push_back(arg); if (!arg_def.type_attr().empty()) { inferred_attrs.insert(arg_def.type_attr()); if (!arg_def.number_attr().empty()) { inferred_attrs.insert(arg_def.number_attr()); } } else if (!arg_def.type_list_attr().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 01 21:05:56 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/model/arg_type.cc
return ArgType(arg_def, kOutput); } ArgType::ArgType(const OpDef::ArgDef& arg_def, Kind kind) : kind_(kind), data_type_(arg_def.type()) { if (!arg_def.type_attr().empty()) { type_attr_name_ = arg_def.type_attr(); } if (!arg_def.type_list_attr().empty()) { type_attr_name_ = arg_def.type_list_attr(); } is_list_ =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 15 18:23:40 UTC 2021 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/convert_type.cc
} } mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr) { auto type = type_attr.getValue(); auto shaped_type = mlir::dyn_cast<mlir::ShapedType>(type); if (shaped_type) { return shaped_type.getElementType(); } else { return type; } } bool NotFromQuantOpOrSameQuantType(mlir::Value val, mlir::TypeAttr qtype_attr) { auto val_defn_op = val.getDefiningOp();
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/lite/transforms/modify_io_nodes.cc
dequantize_input.getType(), dequantize_op.getLoc()); // replace the dequantize op by a quantize op TypeAttr type_attr = TypeAttr::get(returned_type); auto quantize_op = builder.create<QuantizeOp>( dequantize_op.getLoc(), returned_type, dequantize_input, type_attr); returned_value = quantize_op.getOutput(); } else { output_type.print(llvm::errs() << "Requested output type ");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
builder.setInsertionPoint(&block, ++Block::iterator(op)); } else { builder.setInsertionPointToStart(&block); } TypeAttr type_attr = TypeAttr::get(new_type); auto quantize = builder.create<TFL::QuantizeOp>(value.getLoc(), new_type, value, type_attr); auto dequantize = builder.create<TFL::DequantizeOp>( value.getLoc(), expressed_type, quantize.getOutput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
Status status; NodeDef def; def.set_op("Bitcast"); def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr); (*def.mutable_attr())["T"] = typeAttr; (*def.mutable_attr())["type"] = outTypeAttr; def.add_input(
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/lite/utils/constant_utils.cc
#include "tensorflow/core/framework/tensor_shape.pb.h" #include "tensorflow/core/platform/status.h" #include "tsl/platform/statusor.h" namespace mlir { namespace TFL { absl::StatusOr<TypedAttr> CreateTypedAttr(ShapedType shaped_type, int value) { Type element_type = shaped_type.getElementType(); if (element_type.isF16()) { auto floatType = mlir::FloatType::getF16(element_type.getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.5K bytes - Viewed (0)