- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 53 for ret_types (0.16 sec)
-
tensorflow/compiler/jit/xla_kernel_creator.cc
// Create the kernel. Device* dev = flr->device(); Status s; auto props = std::make_shared<NodeProperties>( &fbody->record->fdef().signature(), node_def, fbody->arg_types, fbody->ret_types); OpKernelConstruction construction(DeviceType(dev->device_type()), dev, dev->GetAllocator(AllocatorAttributes()), flr, dev->resource_manager(), props,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
tensorflow::MemoryTypeVector GetOutputMemoryTypes( const tensorflow::FunctionBody* fbody) { tensorflow::MemoryTypeVector output_memory_types(fbody->ret_types.size(), tensorflow::DEVICE_MEMORY); for (size_t i = 0; i < fbody->ret_types.size(); ++i) { if (fbody->ret_types[i] == tensorflow::DT_RESOURCE) { output_memory_types[i] = tensorflow::HOST_MEMORY; } } return output_memory_types;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
llvm::SmallVector<mlir::Type, 4> ret_types; ret_types.reserve(fbody.ret_types.size()); for (auto ret : fbody.ret_nodes) { // Find node in the graph using the node id instead of using `ret` directly // because the graph has been cloned. auto* node = graph_->FindNodeId(ret->id()); TF_ASSIGN_OR_RETURN(auto type, InferInputType(*node, /*idx=*/0, builder)); ret_types.push_back(type); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
DT_RESOURCE, DT_RESOURCE, DT_RESOURCE}), result.arg_types); EXPECT_EQ((DataTypeVector{DT_FLOAT, DT_INT32, DT_FLOAT, DT_FLOAT}), result.ret_types); TF_EXPECT_GRAPH_EQ(expected_body_def, result.gdef); } // Encapsulates the same computation again, verifies we reuse the same // function. Encapsulation should be deterministic to avoid recompilation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
<< type_or_err.status().ToString(); return type_or_err.status(); } auto type = std::move(type_or_err).value(); ret_types.push_back(type); } auto func_type = builder.getFunctionType(input_types, ret_types); // Construct function object auto func = FuncOp::create(func_loc, name, func_type, /* attrs= */ {}); func.addEntryBlock(); auto& body = func.getBody();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_gen.py
tfr_op_name = 'tfr.' + op_name[5:] ret_tys = ( TFR_BUILTINS[op_name](*arg_tys) if callable(TFR_BUILTINS[op_name]) else TFR_BUILTINS[op_name]) # Convert the tfr builtin returns to a list. if isinstance(ret_tys, tuple): ret_tys = list(ret_tys) else: ret_tys = [ret_tys] ret_str, ret_ssa_values = self._get_mlir_ssa_values(op_name, ret_tys) arg_str = ', '.join(arg_strs)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 27 15:27:03 UTC 2022 - 55.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
llvm::SmallVector<Type, 2> new_types(op->getResultTypes()); new_types.back() = rewriter.getType<tf_executor::ControlType>(); llvm::SmallVector<Value, 2> new_operands; FilterOutBlockArgControlDep(operands, new_operands); rewriter.replaceOpWithNewOp<tf_executor::ControlTriggerOp>( op, new_types, new_operands, op->getAttrs()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
platforms/software/dependency-management/src/integTest/groovy/org/gradle/integtests/resolve/RepositoryInteractionDependencyResolveIntegrationTest.groovy
expectGetMetadata() expectGetArtifact() } } repoTypes.subList(repoTypes.indexOf(repoType) + 1, repoTypes.size()).each { other -> "org:$other:1.0" { expectGetMetadataMissingThatIsFoundElsewhere() }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Tue Oct 10 21:10:11 UTC 2023 - 15.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h
namespace mlir { namespace OpTrait { namespace TF { // Verifies if 'ref_type' is a REF type corresponding to 'type'. static inline LogicalResult VerifyRefTypeMatch(mlir::Type type, mlir::Type maybe_ref_type) { if (auto ref_type = mlir::dyn_cast<mlir::tf_type::TensorFlowRefType>(maybe_ref_type)) return success(ref_type.RemoveRef().getTypeID() == type.getTypeID()); return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/types_util.cc
}) .Case<UnknownLoc>([&](UnknownLoc loc) { mutable_location->set_type(ConverterErrorData::UNKNOWNLOC); }) .Case<FileLineColLoc>([&](FileLineColLoc loc) { if (!mutable_location->has_type()) { mutable_location->set_type(ConverterErrorData::CALLSITELOC); } auto new_call = mutable_location->mutable_call()->Add();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.9K bytes - Viewed (0)