- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 345 for setLoc (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
auto min_scaled = rewriter.create<DivOp>(op.getLoc(), float_min, quant_to_float); auto min_scaled_sub = rewriter.create<SubOp>(op.getLoc(), quant_min, min_scaled); auto mid_rounded = rewriter.create<RoundOp>(op.getLoc(), scalar_ty, min_scaled_sub); auto nudged_zero_point_val = rewriter.create<ClipByValueOp>( op.getLoc(), scalar_ty, mid_rounded, quant_min, quant_max);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto range = rewriter.create<mhlo::SubtractOp>(op.getLoc(), limit, start); auto abs = rewriter.create<mhlo::AbsOp>(op.getLoc(), range); // Delta is not necessarily the same type as start and limit. auto abs_cast = rewriter.create<mhlo::ConvertOp>(op.getLoc(), compute_type, abs); auto delta_cast = rewriter.create<mhlo::ConvertOp>(op.getLoc(), compute_type, delta);
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/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
auto tf_reduce_op = rewriter.create<BooleanReduce>( reduce_op.getLoc(), reduce_op->getResult(0).getType(), operand, reduction_indices, /*keep_dim=*/rewriter.getBoolAttr(false)); auto tf_argreduce_op = rewriter.create<ArgReduce>( reduce_op.getLoc(), reduce_op->getResult(1).getType(), operand, reduction_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
filter_constant_op->getLoc(), *rewriter.getContext(), filter_quantized_type.getScale(), filter_quantized_type.getZeroPoint(), /*narrow_range=*/true); } // Required because the quantized dimension is changed from 3 -> 0. auto new_filter_result_type = RankedTensorType::getChecked( filter_constant_op.getLoc(), /*shape=*/new_filter_shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
OpBuilder builder(op); Operation* load = builder.create<mlir::ml_program::GlobalLoadOp>( op.getLoc(), globalOp.getType(), SymbolRefAttr::get(op->getContext(), globalOp.getSymName())); if (globalOp.getType() != op.getValue().getType()) { load = builder.create<TF::CastOp>(op.getLoc(), op.getValue().getType(), load->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
builder.create<ConstOp>(op->getLoc(), output_type, attribute); } else { attr_cst = builder.create<mlir::arith::ConstantOp>( op->getLoc(), cast<TypedAttr>(attribute)); } new_operands.push_back(attr_cst); } } // Create the TFR call op auto new_op = builder.create<CallOp>( op->getLoc(), compose_func_type.getResults(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0), const_op.getResult()}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
ArrayRef<Value> args; builder.create<func::CallOp>(session_initializer.getLoc(), func.getFunctionType().getResults(), func.getSymName(), args); } builder.create<func::ReturnOp>(session_initializer.getLoc()); session_initializer.erase(); } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
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)