- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 173 for getFlow (0.17 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
Type output_type = bmm_op.getResult().getType(); auto no_input = rewriter.create<TFL::NoValueOp>( bmm_op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr()); auto fc_op = rewriter.create<TFL::FullyConnectedOp>( bmm_op->getLoc(), ArrayRef<Type>{output_type}, /*input=*/output_lhs, /*filter=*/output_rhs, /*bias=*/no_input,
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/internal/passes/extract_outside_compilation.cc
builder.create<mlir::TF::YieldOp>(if_region.getLoc(), /*operands=*/ArrayRef<Value>{}); // Create empty else branch region. auto& else_branch = host_side_if.getElseBranch(); else_branch.push_back(new Block); builder.setInsertionPointToEnd(&else_branch.front()); builder.create<mlir::TF::YieldOp>(if_region.getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. { Block &entry_block = outlined_func.getBody().front(); auto loc = outlined_func.getLoc(); for (Value operand : operands) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
} } else { bias = rewriter .create<AddOp>(add_op.getLoc(), bias, constant_val, rewriter.getStringAttr("NONE")) .getOutput(); } auto fc = rewriter.create<TFL::FullyConnectedOp>( FusedLoc::get(fc_op.getContext(), {fc_op.getLoc(), add_op.getLoc()}), add_op.getType(), /*input=*/fc_op.getInput(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
Value scalar_zero = CreateI32SplatConst(op.getLoc(), rewriter, {}, 0); Type shape_dtype = getElementTypeOrSelf(op.getElementShape().getType()); Value num_elements = operands[1]; IntegerAttr attr; if (matchPattern(num_elements, m_Constant(&attr))) { return CreateI32SplatConst(op.getLoc(), rewriter, {1}, attr.getInt()); }
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/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
} operand = rewriter.create<mhlo::BitcastConvertOp>(op->getLoc(), *output_type, operand); Value res_min_clipped = rewriter.create<chlo::BroadcastMaxOp>( op->getLoc(), *output_type, operand, *min_or, broadcast_dims); Value res_max_clipped = rewriter.create<chlo::BroadcastMinOp>( op->getLoc(), *output_type, res_min_clipped, *max_or, broadcast_dims);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
} auto anonymous_iterator = builder.create<AnonymousIteratorV3Op>( reduce_dataset.getLoc(), RankedTensorType::get({}, builder.getType<ResourceType>()), /*output_types=*/builder.getArrayAttr(type_attrs), /*shape_types=*/builder.getArrayAttr(shape_attrs)); builder.create<MakeIteratorOp>(reduce_dataset.getLoc(), reduce_dataset.getInputDataset(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
rewriter.create<arith::ConstantOp>(op->getLoc(), type, begin_attr); auto end_attr = DenseElementsAttr::get<int32_t>(type, padded_end); auto end_op = rewriter.create<arith::ConstantOp>(op->getLoc(), type, end_attr); auto stride_attr = DenseElementsAttr::get<int32_t>(type, padded_stride); auto stride_op = rewriter.create<arith::ConstantOp>(op->getLoc(), type, stride_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
OpBuilder builder(func.getBody()); std::string empty_option_buffer; auto op = builder.create<CustomOp>( func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), api, CustomOption(&builder, empty_option_buffer)); builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); return success(); } LogicalResult VerifyNgrams(func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
assert(caller_region.empty() && "Expected empty region for newly created ops"); OpBuilder builder(caller_region); Block* entry = builder.createBlock(&caller_region); auto loc = op->getLoc(); if (use_region_args) { auto inputs = func.getFunctionType().getInputs(); entry->addArguments(inputs, SmallVector<Location>(inputs.size(), loc)); args = entry->getArguments(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0)