Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 117 for output_types (0.24 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top