- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 1,070 for Auto (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
devices.reserve(device_map.size()); for (auto it : device_map) { auto device_alias = it.getName().strref(); auto device_list = mlir::cast<ArrayAttr>(it.getValue()); llvm::SmallVector<StringRef, 4> device_list_for_alias; device_list_for_alias.reserve(device_list.size()); for (auto device : device_list) device_list_for_alias.emplace_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
// Sink down constants into the functions. for (auto extern_value : region_extern_values) { if (!matchPattern(extern_value, m_Constant())) { extern_values.insert(extern_value); continue; } // Add constant at start of region. auto const_builder = OpBuilder::atBlockBegin(®ion->front()); auto const_value = const_builder.clone(*extern_value.getDefiningOp());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
PatternRewriter& rewriter) { int64_t shape_rank = shape.size(); auto shape_spec_type = RankedTensorType::get({shape_rank}, rewriter.getIntegerType(64)); Type resultType = RankedTensorType::get(shape, element_type); auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); auto shape_tensor = rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
const auto mins = dyn_cast<DenseFPElementsAttr>(min); const auto maxs = dyn_cast<DenseFPElementsAttr>(max); if (mins && maxs) { min_value.reserve(mins.getNumElements()); max_value.reserve(maxs.getNumElements()); for (auto it = mins.begin(); it != mins.end(); ++it) { min_value.push_back(FloatAttr::getValueAsDouble(*it)); } for (auto it = maxs.begin(); it != maxs.end(); ++it) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
} bool checkWhetherGraphHasValidStaticLookupTables(ModuleOp module) { auto hashtables = GetAllOps<TF::HashTableV2Op>(&module); // No needs to run the legalization patterns. if (hashtables.empty()) { return false; } for (auto hashtable : hashtables) { auto key_dtype = hashtable.getKeyDtype(); auto value_dtype = hashtable.getValueDtype();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/device_target.cc
auto in_spec = input_specs[0].dyn_cast<UniformQuantizedType>(); // TODO(fengliuai): handles the PerAxis QuantizedType. auto w_spec = input_specs[1].dyn_cast<UniformQuantizedType>(); auto b_spec = input_specs[2].dyn_cast<UniformQuantizedType>(); auto o_spec = out_specs[0].dyn_cast<UniformQuantizedType>(); if (!in_spec || !w_spec || !b_spec || !o_spec) return failure(); double scale_product = in_spec.getScale() * w_spec.getScale();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
void PopulateTfVersions(mlir::ModuleOp module, const VersionDef& versions) { mlir::Builder b(module.getContext()); auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer())); auto min_consumer = b.getNamedAttr( "min_consumer", b.getI32IntegerAttr(versions.min_consumer())); auto bad_consumers = b.getNamedAttr( "bad_consumers", b.getI32ArrayAttr(llvm::ArrayRef<int32_t>(
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/lite/transforms/push_transpose_through_ewise.cc
} if (!tpose_arg->hasOneUse()) { return failure(); } auto tpose_arg_type = llvm::dyn_cast<RankedTensorType>(tpose_arg->getResultTypes()[0]); auto cst_arg_type = llvm::dyn_cast<RankedTensorType>(cst_arg->getResultTypes()[0]); auto tpose_arg_rank = tpose_arg_type.getRank(); auto cst_arg_rank = cst_arg_type.getRank();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
SP_Stream stream) -> void { auto custom_stream = static_cast<SP_Stream_st*>(stream); ASSERT_EQ(custom_stream->stream_id, 14); delete custom_stream; stream_deleted = true; }; StreamExecutor* executor = GetExecutor(0); ASSERT_FALSE(stream_created); TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream()); ASSERT_TRUE(stream_created);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0)