- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 345 for setLoc (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
auto island = builder.create<IslandOp>( sub_op->getLoc(), sub_op->getResultTypes(), control_type, control_inputs); island.getBody().push_back(new Block); Block* block = &island.getBody().back(); builder.setInsertionPointToEnd(block); sub_op->replaceAllUsesWith(island.getOutputs()); sub_op->moveBefore(block, block->begin()); builder.create<YieldOp>(sub_op->getLoc(), sub_op->getResults()); return island; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
// and its users. Value value = tf_op.getOutputs(); auto quantize = rewriter.create<quantfork::QuantizeCastOp>( tf_op.getLoc(), qtype.getValue(), value); auto dequantize = rewriter.create<quantfork::DequantizeCastOp>( tf_op.getLoc(), res_type, quantize.getResult()); value.replaceAllUsesWith(dequantize); quantize.getOperation()->replaceUsesOfWith(dequantize, value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_filter.cc
filtered_ops.push_back(func); } return; } llvm::Regex op_regex(tac_filter.op_filter().op_name_pattern()); module.walk([&](Operation* op) { auto named_loc = mlir::dyn_cast<NameLoc>(op->getLoc()); if (!named_loc) { return; } if (!op_regex.match(named_loc.getName())) { return; } op->setAttr(kSkipTargetAnnotation, builder.getUnitAttr());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
builder.create<SparseQConstOp>(op->getLoc(), cst.getQtypeAttr(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_qconst.getResult()); cst.erase(); } if (result.needs_densify) { auto value = op->getOperand(operand); auto densify = builder.create<DensifyOp>(op->getLoc(), value.getType(), value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
for (int64_t dim : element_shape) { buffer_shape.push_back(dim); } auto zero = CreateScalarConst(0, builder, op->getLoc()); if (getElementTypeOrSelf(zero.getType()) != element_dtype) { zero = builder.create<TF::CastOp>( op->getLoc(), ArrayRef<Type>{tensorflow::GetTypeFromTFTensorShape({}, element_dtype)}, ArrayRef<Value>{zero}); } auto buffer_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_reorder_replicate_and_partitioned_inputs.cc
replicated_input.getLoc(), replicated_input.getType(), operands_per_replica, replicated_input->getAttrs()); replicate_op.setIsPacked(is_packed); operands_per_core.push_back(replicate_op); } auto pi = builder.create<TF::TPUPartitionedInputV2Op>( first_partitioned_input.getLoc(), replicated_input.getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 24 23:08:55 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization.cc
bool EmitMustBeConstantError(Operation* op) { if (operations_to_skip->contains(op->getRegisteredInfo()->getTypeID())) { IncrementCounterFor(mlir_non_static_op_skip_count, op); return true; } emitError(op->getLoc()) << absl::StrCat( "Node `", op->getName().getStringRef().str(), "` ", kMustBeConstantError); return false; } bool IsStaticOperation(Operation* op) { for (auto o : op->getResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
new_operands[index] = builder.create<TF::ReadVariableOp>( caller->getLoc(), ArrayRef<Type>{new_type}, ArrayRef<Value>{operand}); caller->setOperand(index, new_operands[index]); if (updated_index < 0) continue; builder.setInsertionPointAfter(caller); builder.create<TF::AssignVariableOp>( caller->getLoc(), ArrayRef<Type>{}, ArrayRef<Value>{operand, caller->getResult(updated_index)}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
// It is currently not possible to easily pack the output of a multi-result // op into an op with a single varidic output in `.td`. auto converted = rewriter.create<TFL::CustomOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), "")); rewriter.replaceOp(op, converted.getResults()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
"$_builder.create<TF::ShapeOp>($0.getLoc(), $1, $2)">; def IsI32 : NativeCodeCall< "$_builder.getBoolAttr(getElementTypeOrSelf($0.getType()).isInteger(32))">; def CreateTFCastOpF32 : NativeCodeCall< "CreateTFCastOpF32(&$_builder, $0.getLoc(), $1, $2)">; def CreateTFCastOpI32 : NativeCodeCall< "CreateTFCastOpI32(&$_builder, $0.getLoc(), $1, $2)">; def CreateTensorScatterNdOp : NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0)