- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 129 for created (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
return WalkResult::advance(); }); if (result.wasInterrupted()) { return failure(); } return builder.getArrayAttr(function_list); } // Creates a pruned module containing the XlaCallModule's entry function and // other functions transitively called by the entry function. FailureOr<OwningOpRef<ModuleOp>> PruneStablehloModule(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
} 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()); value.replaceAllUsesWith(dequantize);
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/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
if (failed(applyPatternsAndFoldGreedily(func, frozen_patterns))) { func.emitError() << "quant-propagate-quantize-type failed."; signalPassFailure(); } } } } // namespace // Creates an instance of the TensorFlow dialect PropagateQuantizeType pass. std::unique_ptr<OperationPass<ModuleOp>> CreatePropagateQuantizeTypePass() { return std::make_unique<PropagateQuantizeType>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batch_function_fallback_benchmark_test.cc
#include "tfrt/support/rc_array.h" // from @tf_runtime #include "tfrt/tensor/dense_host_tensor.h" // from @tf_runtime #include "tfrt/tensor/tensor.h" // from @tf_runtime namespace tensorflow { namespace { // Creates a BEF file with a program that runs // tfrt_fallback_async.batch_function with a empty function forwarding inputs or // outputs. std::pair<tfrt::BefBuffer, tfrt::RCReference<tfrt::BEFFile>> CreateBefFile(
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/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
if (constant.getType().getRank() != 2) return failure(); // Create a tfl.transpose op that performs ZX transpose on `input`. auto create_z_x_transpose_op = [&](Value input) -> Value { RankedTensorType input_type = mlir::cast<RankedTensorType>(input.getType()); const int input_rank = input_type.getRank(); // Create a 1D I32 tensor for representing the dimension permutation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets_test.cc
module_(mlir::ModuleOp::create(mlir::UnknownLoc::get(&context_))), builder_(&module_->getBodyRegion()) { context_.loadAllAvailableDialects(); } protected: mlir::MLIRContext context_; mlir::OwningOpRef<mlir::ModuleOp> module_; mlir::OpBuilder builder_; }; TEST_F(XlaLegalizeTargetsTest, CreatesConversionTargets) { auto const_int = builder_.create<mlir::arith::ConstantIntOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
func->removeAttr("tf_saved_model.exported_names"); ArrayRef<Value> args; builder.create<func::CallOp>(session_initializer.getLoc(), func.getFunctionType().getResults(), func.getSymName(), args); } builder.create<func::ReturnOp>(session_initializer.getLoc()); session_initializer.erase(); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
ASSERT_TRUE(tensorflow::GrpcServer::Create( server_def, tensorflow::Env::Default(), &worker_server1) .ok()); ASSERT_TRUE(worker_server1->Start().ok()); server_def.set_task_index(2); std::unique_ptr<tensorflow::GrpcServer> worker_server2; ASSERT_TRUE(tensorflow::GrpcServer::Create( server_def, tensorflow::Env::Default(), &worker_server2)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 22:09:57 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
mlir::OpBuilder::InsertPoint insert_point = builder->saveInsertionPoint(); auto launch = builder->create<mlir::tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new mlir::Block); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<mlir::tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/utils.cc
ConstantOp GetScalarConstOfType(Type ty, Location loc, int64_t raw_value, OpBuilder* builder) { return builder->create<ConstantOp>(loc, hlo::getScalarOfType(ty, raw_value)); } ConstantOp GetScalarNegZeroOfType(Type ty, Location loc, OpBuilder* builder) { return builder->create<ConstantOp>(loc, hlo::getScalarNegZeroOfType(ty)); } DenseIntElementsAttr GetI64ElementsAttr(ArrayAttr attr) { RankedTensorType ty =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.8K bytes - Viewed (0)