- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 159 for getOperands (0.18 sec)
-
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
const int operands_size = while_op.getNumOperands() + extra_operands.size(); SmallVector<Value, 4> operands; operands.reserve(operands_size); operands.append(while_op.getOperands().begin(), while_op.getOperands().end()); operands.append(extra_operands.begin(), extra_operands.end()); SmallVector<Type, 4> new_types; new_types.reserve(operands_size); new_types.append(while_op.getResultTypes().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
decomposed_partitioned_call_callees))) { return failure(); } if (!signature_change) return success(); auto new_if_operands = llvm::to_vector<8>(if_op.getOperands()); for (auto operand : if_op.getOperands()) { auto it = data_var_to_size_var.find(operand); if (it == data_var_to_size_var.end()) continue; new_if_operands.push_back(it->getSecond()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
}; if (!llvm::all_of(operation->getOperands(), foldable_opr)) { return std::nullopt; } return FoldAdaptor(operation); } // Gets a list of ElementsAttr behind each constant operand. llvm::SmallVector<ElementsAttr> OperandData() { llvm::SmallVector<ElementsAttr> res; res.reserve(operation_->getNumOperands()); for (auto opr : operation_->getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
PatternRewriter &rewriter) const final { Operation *op = srcop.getOperation(); bool allTypesFp = true; bool allTypesQuantizedOrInt = true; for (auto operand : op->getOperands()) { ShapedType type = mlir::dyn_cast<ShapedType>(operand.getType()); if (!type) continue; allTypesFp &= !mlir::isa<quant::QuantizedType>(type.getElementType()); allTypesQuantizedOrInt &=
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h
*count = output_type.getNumElements(); return true; } static bool GetInputTensorTotalSize(mlir::Operation* op, int64_t* count) { int64_t total_count = 0; for (auto input : op->getOperands()) { auto input_type = mlir::dyn_cast_or_null<mlir::RankedTensorType>(input.getType()); if (!input_type || !input_type.hasStaticShape()) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto old_terminator = block.getTerminator(); auto new_outputs = llvm::to_vector<8>(old_terminator->getOperands()); llvm::SmallVector<std::tuple<int64_t, int64_t, bool>, 8> output_buffer_to_size; for (auto retval : llvm::enumerate(old_terminator->getOperands())) { auto it = buffer_to_size.find(retval.value()); if (it == buffer_to_size.end()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc
composite->getLoc(), composite->getResultTypes(), composite->getOperands().slice(2, 3), composite.getName(), CustomOption(&builder, custom_option_buffer)); } return builder.create<TFL::CustomOp>( composite->getLoc(), composite->getResultTypes(), composite->getOperands(), composite.getName(), CustomOption(&builder, custom_option_buffer)); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc
// Get total bytes transferred. int64_t GetTransferredTensorBytes(func::CallOp from_graph, func::CallOp to_graph) { int64_t total_size_transferred = 0; for (auto input : to_graph.getOperands()) { Operation* input_op = input.getDefiningOp(); if (input_op && input_op == from_graph.getOperation()) { auto input_type = mlir::dyn_cast_or_null<RankedTensorType>(input.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
YieldOp yield = GetYield(); return wrapped_op.getNumResults() == yield.getNumOperands() && std::equal(wrapped_op.getResults().begin(), wrapped_op.getResults().end(), yield.getOperands().begin()); } mlir::LogicalResult IslandOp::verify() { IslandOp island = *this; if (!island.GetBody().args_empty()) return island.emitOpError() << "expects body without any arguments";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
op.getLoc(), op.getType(), op.getOperands(), builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false), op.get_XlaShardingAttr()); op->replaceAllUsesWith(pi); } else { auto po = builder.create<TF::TPUPartitionedOutputV2Op>( op.getLoc(), op.getResultTypes(), op.getOperand(), builder.getI64ArrayAttr(partition_dims), op.get_XlaShardingAttr());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0)