- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 380 for getLoc (0.12 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
return failure(); } auto tfl_fusable_op = builder.create<TFL::CustomOp>( func->getLoc(), func.getFunctionType().getResults(), func.getArguments(), custom_op_name, CustomOption(&builder, custom_option_buffer)); builder.create<func::ReturnOp>(func->getLoc(), tfl_fusable_op.getResults()); return success(); } // Abstracts the conversion of the embedded lookup composite function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc
if (!dict_attr) continue; OpBuilder builder(func.getContext()); builder.setInsertionPointToStart(&func.getBlocks().front()); builder.create<TFL::CallOnceOp>(func.getLoc(), init_func_op.getName()); } } } } // namespace // Inserts a TFL::CallOnce op when tf_saved_model's session initializer is // given. std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc
operand.getDefiningOp<corert::ConstDenseTensorOp>()) { new_values.push_back( rewriter.create<fallback_async::ConstDenseTensorOp>( op.getLoc(), rewriter.getType<fallback::TFTensorType>(), corert_const_dense_tensor_op.getValue())); should_rewrite = true; continue; } if (auto corert_const_string_tensor_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
auto restore_op = builder.create<TF::RestoreV2Op>( session_init_func.getLoc(), /*tensors=*/tensor_types, /*prefix=*/filename_arg, tensor_names_const, shape_and_slices_const); for (auto [idx, restore_result] : llvm::enumerate(restore_op.getResults())) { builder.create<TF::AssignVariableOp>( restore_op.getLoc(), target_var_handle_ops[idx], restore_result); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
auto merged_execute_launch = builder->create<tf_device::LaunchOp>( merged_execute.getLoc(), execute_launch.getDeviceAttr(), merged_execute.getResultTypes()); merged_execute_launch.getBody().push_back(new Block); builder->setInsertionPointToEnd(&merged_execute_launch.GetBody()); builder->create<tf_device::ReturnOp>(merged_execute.getLoc(), merged_execute.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
transposer.TransposeValues(original_values); // Create a new constant op with the transposed values. const Location combined_loc = rewriter.getFusedLoc({const_op.getLoc(), op.getLoc()}); auto new_value_type = RankedTensorType::getChecked(combined_loc, transposer.GetTargetShape(), /*elementType=*/rewriter.getF32Type()); auto new_value_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/utils.h
Block* block = builder->createBlock(body); // Block arguments are scalars of the given element type. Type type = RankedTensorType::get(/*shape=*/{}, element_type); Location loc = body->getLoc(); block->addArguments({type, type}, SmallVector<Location, 2>(2, loc)); auto reducer = builder->create<Op>(loc, block->getArgument(0), block->getArgument(1)); builder->create<ReturnOp>(loc, reducer.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
if (existing) return *existing; } ModuleOp module = region.getParentOfType<ModuleOp>(); OpBuilder builder(module.getContext()); auto loc = region.getParentOp()->getLoc(); Block& entry = region.front(); int num_region_arguments = entry.getNumArguments(); Operation* terminator = entry.getTerminator(); // Build the function type. Region arguments and extern values together
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/quantization/stablehlo/passes/quantization_patterns.cc
SmallVector<Type> arg_types; SmallVector<Location> arg_locs; for (const Value arg : xla_call_module_op.getArgs()) { arg_types.push_back(arg.getType()); arg_locs.push_back(arg.getLoc()); } SmallVector<Type> output_types; for (const Value output : xla_call_module_op.getOutput()) { output_types.push_back(output.getType()); } entry_func_op.setFunctionType(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0)