- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for MutableArrayRef (0.59 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
xla::XlaComputation* xla_computation, bool use_tuple_args, bool enable_op_fallback, bool return_tuple, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns = {}, llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes = {}, llvm::StringRef module_name = llvm::StringRef()); // Creates a MLIR pipeline that lowers MLIR module to MHLO dialect. The input
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
// true then the pipeline will include all the legalization passes. void CreateConvertMlirToXlaHloPipeline( mlir::OpPassManager& pm, llvm::StringRef device_type, bool enable_op_fallback, llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes, bool lower_to_xla_hlo, bool allow_partial_conversion) { bool legalize_chlo = true; pm.addNestedPass<mlir::func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
const TF::ResourceAliasAnalysis::Info& resource_alias_analysis) { // We need to know the devices to copy to. if (!replicate.getDevices()) return false; MutableArrayRef<OpOperand> inputs = replicate.GetOperandsForBlockArgument(replicate_arg); for (const auto& entry : llvm::enumerate(inputs)) { auto input_op = entry.value().get().getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
// // Requires that block argument is of this replicate op. MutableArrayRef<OpOperand> ReplicateOp::GetOperandsForBlockArgument( BlockArgument block_arg) { assert(block_arg.getOwner() == &GetBody()); unsigned arg_number = block_arg.getArgNumber(); unsigned num_replicated_args = GetNumReplicatedBlockArguments(); int32_t num_replicas = getNAttr().getInt(); MutableArrayRef<OpOperand> operands = getOperation()->getOpOperands();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
Status CompileMlirToXlaHloViaBuilder( mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, llvm::StringRef device_type, XlaCompilationResult* compilation_result, llvm::MutableArrayRef<std::unique_ptr<mlir::Pass>> custom_legalization_passes) { // This call to RefineShapes is redundant with the call in BuildHloFromTf. // It's here so xla::Parameters that are created form block.getArguments will
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
unsigned GetReplicaOperandIndexForBlockArgument(BlockArgument block_arg, unsigned replica); Value GetReplicaOperandForBlockArgument(BlockArgument block_arg, unsigned replica); MutableArrayRef<OpOperand> GetOperandsForBlockArgument(BlockArgument block_arg); bool WrapsSingleOp(); }]; let builders = [ OpBuilder<(ins "int":$n,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
return new_island; } // Creates respective YieldOp for the new merged island. YieldOp CreateNewIslandYieldOp(IslandOp new_island, llvm::MutableArrayRef<IslandResult> results) { llvm::SmallVector<Value, 8> yield_operands; yield_operands.reserve(results.size()); for (auto [old_result, new_island] : llvm::zip(results, new_island.getOutputs())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
return mapping; } // Adds a new replicated input to the replicate op. tf_device::ReplicateOp AddInputsToReplicateOp( tf_device::ReplicateOp replicate, MutableArrayRef<TF::VarHandleOp> new_inputs, const llvm::SmallDenseMap<llvm::StringRef, llvm::SmallVector<StringRef, 4>>& devices) { int64_t num_replicas = replicate.getN(); assert(new_inputs.size() == num_replicas);
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/tpu_space_to_depth_pass.cc
tf_device::ReplicateOp replicate, int32_t block_size) { // We need to know the devices to copy to. if (!replicate.getDevices()) return false; MutableArrayRef<OpOperand> inputs = replicate.GetOperandsForBlockArgument(block_arg); for (auto& input : inputs) { auto input_op = input.get().getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
LLVM_DEBUG(llvm::dbgs() << GemmStyleOp::getOperationName() << " op must have ranked tensor type.\n"); return failure(); } MutableArrayRef<BlockArgument> operands = entry_func_op.getBody().getArguments(); // Function must have input, filter, and optionally bias. if (operands.size() != 2 && operands.size() != 3) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0)