- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for GetOperation (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>(); auto symbol = mlir::SymbolTable::lookupNearestSymbolFrom(call.getOperation(), sym); if (!symbol) return failure(); auto f = llvm::dyn_cast<mlir::func::FuncOp>(symbol); if (call.getOperation()->getNumOperands() != f.getBody().front().getNumArguments()) { return failure(); // RemoteCall et al }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
// The passes to run on the module. std::vector<GraphOptimizationPass*> passes_; }; } // anonymous namespace void GraphOptPass::runOnOperation() { mlir::ModuleOp module_in = getOperation(); mlir::MLIRContext& ctx = getContext(); // Convert MLIR to Graph FunctionLibraryDefinition flib_def(OpRegistry::Global(), FunctionDefLibrary()); GraphExportConfig confs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
if (output_shape.value()) { changed = RefineResultType(op.getOperation(), op.getResult(), output_shape.value()); return changed; } } return changed; } bool ShapeInference::RefineWithInferTypeOpInterface( InferTypeOpInterface infer_ti) { Operation* op = infer_ti.getOperation(); if (none_of(op->getResultTypes(), CanBeRefined)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
ParseResult $cppClass::parse(OpAsmParser &parser, OperationState &result) { return parseOneResultSameOperandTypeOp(parser, result); } void $cppClass::print(OpAsmPrinter &p) { return printOneResultOp(getOperation(), p); } }]; let hasOptions = 1; } def TFL_AddNOp : TFL_Op<"add_n", [ Commutative, Pure, SameOperandsAndResultsScale,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
operand.assign(orig_to_new[operand.get()]); } } return WalkResult::advance(); }); return success(); } void XlaBroadcast::runOnOperation() { FuncOp func = getOperation(); mlir::ModuleOp module = func->getParentOfType<mlir::ModuleOp>(); if (!module) return signalPassFailure(); func.walk([&](ClusterOp cluster) { if (auto replicate = cluster->getParentOfType<ReplicateOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
const std::vector<int32_t>& operands, const std::vector<int32_t>& results) { std::string op_name = gather_op.getOperation()->getName().getStringRef().str(); uint32_t opcode_index = GetOpcodeIndex(op_name, tflite::BuiltinOperator_STABLEHLO_GATHER); std::vector<int64_t> offset_dims_vec(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
} return WalkResult::advance(); }); return res.wasInterrupted(); } void LowerStaticTensorListPass::runOnOperation() { auto *context = &getContext(); ModuleOp module = getOperation(); // When allow_tensorlist_pass_through_ == false the target dynamic legal // checks will cause this pass to fail if there are any variant type tensors
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
stride_values); auto d_slice = rewriter.create<mhlo::RealDynamicSliceOp>( loc, op.getOperation()->getResult(0).getType(), input, start_indices, end_indices, stride_indices); rewriter.replaceOp(op, d_slice.getOperation()->getResults()); return success(); } }; static void BroadcastBatchMatMulV2Operands(Value lhs, Value rhs, Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
// has two elements, so it will be merging two operand ranges (the two // variadic arguments). Attributes like `batch_timeout_micros` will not // make it to the return array. auto segment_sizes = getOperation()->getAttrOfType<DenseI32ArrayAttr>(getOperandSegmentSizeAttr()); return OperandRangeRange(getOperands(), segment_sizes).join(); } }]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)