- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 83 for ArrayAttr (0.34 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
DictionaryAttr arg_attrs = main_func_op.getArgAttrDict(/*index=*/0); EXPECT_THAT(arg_attrs.get("tf_saved_model.bound_input"), IsNull()); const ArrayRef<Attribute> index_path_attrs = mlir::cast<ArrayAttr>(arg_attrs.get("tf_saved_model.index_path")) .getValue(); EXPECT_THAT(index_path_attrs, SizeIs(1)); StringAttr index_path = mlir::dyn_cast_or_null<StringAttr>(index_path_attrs[0]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
// Set the insertion point right before the return op. builder.setInsertionPoint(®ion.back().back()); StringAttr output_file_path_attr = builder.getStringAttr(output_file_path); ArrayAttr ids_attr = builder.getStrArrayAttr(ids); ArrayAttr calibration_methods_attr = builder.getI32ArrayAttr(calibration_methods); builder.create<TF::CalibrationStatisticsSaverOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
>(); } static LogicalResult VerifyIndexPath(Operation *op, NamedAttribute named_attr) { auto attr = mlir::dyn_cast<ArrayAttr>(named_attr.getValue()); if (!attr) { return op->emitError() << "'" << kTfSavedModelIndexPathAttr << "' attribute should be an ArrayAttr"; } for (auto element : attr) { if (mlir::isa<StringAttr>(element)) { continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
const int func_input_size = func_inputs.size(); const int argument_size = reduce_dataset->getAttrOfType<ArrayAttr>("Targuments").size(); const int state_size = reduce_dataset->getAttrOfType<ArrayAttr>("Tstate").size(); const int dataset_input_size = func_input_size - state_size - argument_size; SmallVector<Type, 2> dataset_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/utils.cc
} ConstantOp GetScalarNegZeroOfType(Type ty, Location loc, OpBuilder* builder) { return builder->create<ConstantOp>(loc, hlo::getScalarNegZeroOfType(ty)); } DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) { RankedTensorType ty = RankedTensorType::get(static_cast<int64_t>(attr.size()), IntegerType::get(attr.getContext(), 64)); return DenseIntElementsAttr::get(ty, attr.getValue()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
// contraction, with two additions: the list of ops which have been fused // together; epsilon (only with FusedBatchNorm). std::vector<NamedAttribute> attrs = contraction->getAttrs(); ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops); attrs.push_back( NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr)); // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
main_func.setArgAttr( i, kTfSavedModelIndexPathAttr, ArrayAttr::get(context, {StringAttr::get(context, input_names[i])})); } const int num_results = main_func.getNumResults(); for (int i = 0; i < num_results; ++i) { main_func.setResultAttr( i, kTfSavedModelIndexPathAttr, ArrayAttr::get(context, {StringAttr::get(context, output_names[i])})); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
version_attr.get("min_consumer")); if (min_consumer) versions->set_min_consumer(min_consumer.getInt()); auto bad_consumers = mlir::dyn_cast_or_null<mlir::ArrayAttr>( version_attr.get("bad_consumers")); if (!bad_consumers) return mlir::success(); for (auto bad_consumer : bad_consumers) { auto bad_consumer_int_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td
// attribute. def TF_SymbolRefArrayAttr : TypedArrayAttrBase<SymbolRefAttr, "tensorflow symbol ref array attribute"> { let constBuilderCall = "::mlir::ArrayAttr::get($_builder.getContext(), $0)"; } //===----------------------------------------------------------------------===// // TensorFlow type definitions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 30.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
shape_attrs.push_back( tf_type::ShapeAttr::get(ctx, mlir::cast<ShapedType>(result_type))); } const auto empty_array_attr = ArrayAttr::get(ctx, {}); // TODO: b/310291615 - find a better way for platform support. const auto platforms = ArrayAttr::get( ctx, {StringAttr::get(ctx, kPlatformCpu), StringAttr::get(ctx, kPlatformTpu)});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0)