- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 117 for output_types (0.24 sec)
-
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, &output_dtys)); mlir::MLIRContext* context = tfr_module_.getContext(); llvm::SmallVector<mlir::Type, 4> input_tys, output_tys; mlir::Builder builder(context); for (auto ty : input_dtys) { mlir::Type elt_ty; TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
staging/src/k8s.io/apiserver/pkg/admission/plugin/cel/compile.go
return resultError("program instantiation failed: "+err.Error(), apiservercel.ErrorTypeInternal) } return CompilationResult{ Program: prog, ExpressionAccessor: expressionAccessor, OutputType: ast.OutputType(), } } func mustBuildEnvs(baseEnv *environment.EnvSet) variableDeclEnvs { requestType := BuildRequestType() namespaceType := BuildNamespaceType()
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 10 22:07:40 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
public: using OpRewritePattern<TF::HashTableV2Op>::OpRewritePattern; LogicalResult matchAndRewrite(TF::HashTableV2Op hashtable_op, PatternRewriter& rewriter) const override { auto output_type = RankedTensorType::get( {1}, TF::ResourceType::get(rewriter.getContext())); // Hash the shared name to generate integer hash table id. The TFLite
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
custom_op.getCustomCode().str()); op_state.addOperands(custom_op.getOperands()); llvm::SmallVector<mlir::Type, 4> output_tys; for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); } op_state.addTypes(output_tys); op_state.addAttributes(attr); auto stablehlo_op = builder.create(op_state); custom_op.replaceAllUsesWith(stablehlo_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc
} Status ConvertToTensorProto(const ElementsAttr attr, TensorProto* output) { auto type = attr.getShapedType(); auto shape = type.getShape(); DataType output_dtype; TF_RETURN_IF_ERROR(ConvertToDataType(type, &output_dtype)); output->set_dtype(output_dtype); ConvertToTensorShapeProto(shape, output->mutable_tensor_shape()); if (auto tensor_attr = mlir::dyn_cast<mlir::TF::TensorProtoAttr>(attr))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 20.5K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass.cc
n->def(), &input_mtypes, &output_mtypes)); for (const Edge* e : n->out_edges()) { Node* dst = e->dst(); if (e->IsControlEdge()) { continue; } bool edge_incurs_extra_device_to_host_copy; if (output_mtypes[e->src_output()] == DEVICE_MEMORY) { // If the output of the *TensorFlow* operation is in DEVICE_MEMORY then
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
!bmm_op.getAdjY() ? create_z_x_transpose_op(input_rhs) : input_rhs; 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/tensorflow/ir/tf_ops_a_m.cc
crops_values.push_back(crops_value_int); } } auto output_type = mlir::cast<TensorType>(op.getOutput().getType()); if (output_type.hasRank()) { if (output_type.getRank() != 4) return op.emitOpError() << "requires output to be a 4D tensor, but got " << output_type; auto static_dims = [](int64_t dim_a, int64_t dim_b) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
platforms/software/reporting/src/main/java/org/gradle/api/reporting/internal/TaskGeneratedSingleDirectoryReport.java
@Nullable private final String relativeEntryPath; @Inject public TaskGeneratedSingleDirectoryReport(String name, Task task, @Nullable String relativeEntryPath) { super(name, OutputType.DIRECTORY, task); this.relativeEntryPath = relativeEntryPath; getOutputLocation().convention(getProjectLayout().dir(new DefaultProvider<>(() -> {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Mar 25 18:51:39 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
Node* var_node; TF_RETURN_IF_ERROR(n->input_node(0, &var_node)); const_def.set_name( graph->NewName(absl::StrCat("var_shape_", var_node->name()))); DataType dtype = n->output_type(0); AddNodeAttr("dtype", dtype, &const_def); TensorProto value; value.set_dtype(dtype); value.mutable_tensor_shape()->add_dim()->set_size( shape_proto.dim_size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0)