- Sort Score
- Result 10 results
- Languages All
Results 201 - 210 of 263 for created (0.14 sec)
-
tensorflow/compiler/jit/device_compilation_profiler.cc
bool used_persistent_cache) { metrics::UpdateXlaCompilationTime(compile_time_us); const std::string& function_name = function.name(); mutex_lock lock(mu_); // Create a stats entry if it doesn't already exist. auto it = cluster_compile_stats_.emplace(function.name(), ClusterCompileStats{}) .first; const uint64 compile_time_s = compile_time_us / 1.0e6;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_ops.cc
core::RefCountPtr<Var> variable; const Tensor& value = context->input(1); // Note: every resource-variable-manipulating op assumes copy-on-write // semantics, and creates a copy of the variable's Tensor if its refcount is // bigger than 1 when we try to modify it. This means we never need to copy // the original tensor for AssignVariableOp; even if there are other live
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_pass.cc
std::move(patterns)))) { getOperation().emitError("Composite lowering pass failed."); signalPassFailure(); } } } // namespace // Creates an instance of the pass. std::unique_ptr<OperationPass<ModuleOp>> CreateCompositeLoweringPass() { return std::make_unique<CompositeLoweringPass>(); } // Registers the pass implementation
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/renderers/op_implementation_renderer.cc
} else if (op_.NumOutputs() == 1) { RenderExecutionSingleOutput(); } else { RenderExecutionMultipleOutputs(); } } void OpImplementationRenderer::RenderInitialization() { // Create Op variable and initialize it Statement("AbstractOperationPtr $0(ctx->CreateOperation())", op_.VariableName()); TFStatement(Call(op_.VariableName(), "Reset",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 05:51:40 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
} FunctionDefLibrary CreateFunctionDefLibWithConstFunction(const string& name) { FunctionDefLibrary fdef_lib; FunctionDef func = FunctionDefHelper::Create( /*function_name=*/name, /*in_def=*/{}, /*out_def=*/{"out: float"}, /*attr_def*/ {}, /*node_def=*/{FunctionDefHelper::Const("one", 1.0f)}, /*ret_def=*/{{"out", "out:output:0"}});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
bcast_dims); } else { return rewriter.notifyMatchFailure(bcast_op, "Unsupported element type."); } Value new_const_op = rewriter.create<mhlo::ConstantOp>(bcast_op.getLoc(), result); rewriter.replaceOp(bcast_op, {new_const_op}); return success(); } }; using FoldBroadcastInDimBeforeMulOp =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
if (!tensor_attr_or.ok()) return nullptr; return tensor_attr_or.value(); } // Creates a constant op that holds 'tensor_elements'. TF::ConstOp GetConstOpFromElementsAttr(ElementsAttr tensor_elements, OpBuilder builder, Location loc) { return builder.create<TF::ConstOp>(loc, tensor_elements); } // Returns ElementsAttr which has the value held by 'resource_tensor'.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
// context in tensorflow_accelerator_device_info(). Also, // tensorflow_accelerator_device_info() == nullptr is used as an IsCPU test. // We need XlaCpuDevice to be treated not as CPU because it allocates // XlaTensors, not regular Tensors. Status status = device->UseAcceleratorDeviceInfo(); if (!status.ok()) { errors::AppendToMessage(&status, "while setting up ", DEVICE_GPU_XLA_JIT);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc
return op->emitError("expected exactly one called_computation"); auto decomposition = mlir::cast<FlatSymbolRefAttr>(calledComputations[0]); auto composite = rewriter.create<mlir::stablehlo::CompositeOp>( op.getLoc(), op.getResultTypes(), op.getOperands(), name.str(), attrs, decomposition.getValue()); rewriter.replaceOp(op, composite.getResults()); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
llvm::StringRef device) { builder->setInsertionPoint(op); auto launch = builder->create<tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new Block); op->replaceAllUsesWith(launch); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0)