- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 330 for setLoc (0.28 sec)
-
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) -
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)