- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 155 for created (0.32 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
return failure(); } // Compute multiplier = scale / sqrt(variance + epsilon) Value multiplier = rewriter.create<mhlo::AddOp>( bn_op.getLoc(), bn_op.getVariance(), epsilon); multiplier = rewriter.create<mhlo::RsqrtOp>(bn_op.getLoc(), multiplier); multiplier = rewriter.create<mhlo::MulOp>(bn_op.getLoc(), multiplier, bn_op.getScale());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
absl::StrCat("device: ", DEVICE_TPU_SYSTEM, " device")); devices->push_back(std::make_unique<VirtualDevice>(options.env, attrs)); VLOG(1) << "Created TPU_SYSTEM device. This host has " << device_count << " TPUs"; return absl::OkStatus(); } } // namespace void RegisterTpuDeviceToDeviceCopy() { static auto* const register_tpu_tpu_copy = new CopyTensor::Registration(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
if (tpu_replicated_input.getIsMirroredVariable()) { mirrored_variable_indices.push_back(pos_and_input.index()); } } // Create replicate op. auto result_types = GetClusterResultTypes(cluster, partitioned_outputs); auto replicate_op = builder.create<mlir::tf_device::ReplicateOp>( cluster.getLoc(), num_replicas, llvm::SmallDenseMap<llvm::StringRef,
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/quantization/tensorflow/passes/insert_main_function.cc
// function, which might not exist in case of multi-signature graphs. In that // case, this pass will create a new main function, which calls signature // functions. // // An already existing @main function will be renamed by attaching a numeric // suffix like `@main_0` to avoid conflict with the newly created main function. class InsertMainFunctionPass : public PassWrapper<InsertMainFunctionPass, OperationPass<ModuleOp>> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
// Test to ensure that a registered graph optimization pass is only executed // once (i.e., on the main function side) in running distributed functions. // This test creates a cluster with two workers, create a variable on the // second worker, and run a distributed function (VariableAddFunction) whose ops // span the local and remote workers. If the graph optimization pass is executed
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
llvm::SmallVector<Value, 2> new_operands; FilterOutBlockArgControlDep(op->getOperands().drop_front(), new_operands); auto source_op = builder.create<tf_executor::NextIterationSourceOp>( op->getLoc(), op->getOperand(0).getType()); builder.create<tf_executor::NextIterationSinkOp>( op->getLoc(), source_op.getToken(), /*input=*/op->getOperand(0), /*controlInputs=*/new_operands);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_cfg.cc
// control flow op into an i1 value. static Value LowerCondition(Location loc, Value value, OpBuilder* builder) { Value zero_d = builder->create<ToBoolOp>(loc, value); Value scalar = builder->create<tensor::ExtractOp>(loc, zero_d); return scalar; } // Calls the function `fn` with arguments provided by the given function and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 13 11:42:59 UTC 2023 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
indices[i] = dimension_numbers.start_index_map()[i]; } // Fill elements from start_indices with start_index_map Value scattered_start_indices = builder.create<TF::TensorScatterUpdateOp>( loc, empty_start_indices, /*indices=*/ builder.create<TF::ReshapeOp>( loc, RankedTensorType::get({index_map_size, 1}, builder.getI64Type()), Create1DConstValue<int64_t>(builder, loc, indices),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
Type resultType = RankedTensorType::get(shape, element_type); auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); auto shape_tensor = rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr); return rewriter.create<TF::ReshapeOp>(loc, resultType, /*tensor=*/value, /*shape=*/shape_tensor); } template <typename BatchMatMulOpType>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils.cc
func->setAttr(kTFImplements, attr); OpBuilder builder(func.getBody()); std::string empty_option_buffer; auto op = builder.create<CustomOp>( func.getLoc(), func.getFunctionType().getResults(), func.getArguments(), api, CustomOption(&builder, empty_option_buffer)); builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); return success(); } LogicalResult VerifyNgrams(func::FuncOp func) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.4K bytes - Viewed (0)