- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 986 for Auto (0.06 sec)
-
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
func.walk([&](Operation* op) { b.setInsertionPoint(op); if (auto dq = llvm::dyn_cast<DequantizeOp>(op)) { auto dcast = b.create<quantfork::DequantizeCastOp>( dq.getLoc(), dq.getOutput().getType(), dq.getInput()); dq.getOutput().replaceAllUsesWith(dcast); dq.erase(); } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) { auto qcast = b.create<quantfork::QuantizeCastOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/cc/experimental/libtf/runtime/runtime.cc
const SavedObjectGraph object_graph = tf_package->GetObjectGraph(); auto& nodes = object_graph.nodes(); // Get a map of the concrete functions to their input / output signatures. auto& concrete_functions = object_graph.concrete_functions(); auto& root = nodes.at(0); for (auto& child : root.children()) { // The child's name describes the name of the edge that connects to the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 11:18:25 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/cc/client/client_session_test.cc
int num_threads = 3; TensorShape data_shape({1, 1}); auto a = Placeholder(root, DT_INT32, Placeholder::Shape(data_shape)); auto b = Placeholder(root, DT_INT32, Placeholder::Shape(data_shape)); auto c = BatchMatMul(root, a, b); ClientSession session(root); std::vector<Tensor> outputs; auto inter_op_threadpool = absl::make_unique<CustomThreadPoolImpl>(num_threads);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 9.5K 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/tensorflow/analysis/tf_dataflow.h
ArrayRef<StateT *> results) { if (auto cast = dyn_cast<TF::CastOp>(op)) { this->join(results[0], *operands[0]); } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) { for (auto ®ion : while_op->getRegions()) { for (auto [arg, value] : llvm::zip(region.getArguments(), while_op->getOperands())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batch_function_fallback_benchmark_test.cc
} TEST(BatchFunctionTest, Basic) { auto corert = CreateTestCoreRuntime(); auto* host = corert->GetHostContext(); auto [bef_buffer, bef_file] = CreateBefFile(host); auto* func = bef_file->GetFunction("main"); CHECK(func); CHECK_EQ(func->result_types().size(), 113); CHECK_EQ(func->argument_types().size(), 113); auto arguments = CreateTestArguments(func, host);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 08 08:08:48 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/cc/gradients/resource_variable_grad_test.cc
TensorShape shape({}); auto scope = Scope::NewRootScope(); auto x = Placeholder(scope, DT_FLOAT, Placeholder::Shape(shape)); auto var = VarHandleOp(scope, DT_FLOAT, shape); auto init = AssignVariableOp(scope, var, Const(scope, 2.0f, shape)); auto temp = ReadVariableOp(scope, var, DT_FLOAT); auto y = Mul(scope, temp, x); auto dy = Placeholder(scope, DT_FLOAT, Placeholder::Shape(shape));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 14 15:30:48 UTC 2021 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
func::FuncOp createMaxUnpoolingFunc( mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape, const SmallVector<int64_t, 4>& output_shape) { auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); SmallVector<mlir::Type, 2> input_types{input_type, indices_type};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0)