Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 31 for argTypes (0.18 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc

                                  func::FuncOp entry_func_op,
                                  TF::XlaCallModuleOp xla_call_module_op) {
      SmallVector<Type> arg_types;
      SmallVector<Location> arg_locs;
      for (const Value arg : xla_call_module_op.getArgs()) {
        arg_types.push_back(arg.getType());
        arg_locs.push_back(arg.getLoc());
      }
    
      SmallVector<Type> output_types;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 06:04:36 UTC 2024
    - 41.7K bytes
    - Viewed (0)
  2. src/text/template/exec.go

    		}
    		return v
    	}
    
    	// Special case for builtin and/or, which short-circuit.
    	if isBuiltin && (name == "and" || name == "or") {
    		argType := typ.In(0)
    		var v reflect.Value
    		for _, arg := range args {
    			v = s.evalArg(dot, argType, arg).Interface().(reflect.Value)
    			if truth(v) == (name == "or") {
    				// This value was already unwrapped
    				// by the .Interface().(reflect.Value).
    				return v
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri May 24 21:22:24 UTC 2024
    - 32K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

    };
    
    // Updates func's type according to its current arguments and return values.
    void UpdateFuncType(func::FuncOp func) {
      llvm::SmallVector<Type, 8> arg_types;
      for (auto arg : func.getArguments()) arg_types.push_back(arg.getType());
      func.setType(
          FunctionType::get(func.getContext(), arg_types,
                            func.front().getTerminator()->getOperandTypes()));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/kernels/xla_ops.cc

      OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(),
                            ctx->GetAttr("Tconstants", &constant_types));
    
      DataTypeVector arg_types;
      OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(),
                            ctx->GetAttr("Targs", &arg_types));
    
      int num_resources;
      OP_REQUIRES_OK_RETURN(ctx, std::vector<int>(),
                            ctx->GetAttr("Nresources", &num_resources));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 22:46:36 UTC 2024
    - 41.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

        ret_operands.push_back(operand->getValue());
      }
      builder_.create<func::ReturnOp>(func_.getLoc(), ret_operands);
    
      auto arg_types = body.getArgumentTypes();
      auto result_types = body.getTerminator()->getOperandTypes();
      func_.setType(FunctionType::get(func_.getContext(), arg_types, result_types));
      *f = new MlirFunction(std::move(context_), std::move(module_), func_);
      return absl::OkStatus();
    }
    
    extern "C" {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc

    };
    
    // Updates func argument type to have the updated input shape.
    void UpdateFuncType(func::FuncOp func) {
      auto arg_types = func.front().getArgumentTypes();
      auto result_types = func.front().getTerminator()->getOperandTypes();
      func.setType(FunctionType::get(func.getContext(), arg_types, result_types));
    }
    
    void HandleFuncOp(Operation* op) {
      auto func = llvm::cast<func::FuncOp>(op);
      UpdateFuncType(func);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 29.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc

      auto init_func = builder.create<func::FuncOp>(
          loc, /*sym_name=*/"init_func_restore_op", func_type);
      builder.createBlock(&init_func.getBody(), /*insertPt=*/init_func.begin(),
                          /*arg_types=*/{}, /*arg_locs=*/{});
    
      init_func->setAttr(kTfSavedModelExportedNamesAttr,
                         builder.getStrArrayAttr(
                             {"tf_saved_model.session_initializer_restore_op"}));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 16 15:04:53 UTC 2023
    - 14K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc

        EXPECT_EQ((DataTypeVector{DT_INT32, DT_FLOAT, DT_INT32, DT_FLOAT,
                                  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);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 16 18:03:15 UTC 2023
    - 14.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

    }
    
    ParseResult TFRFuncOp::parse(OpAsmParser &parser, OperationState &result) {
      auto build_func_type =
          [](Builder &builder, ArrayRef<Type> arg_types, ArrayRef<Type> results,
             function_interface_impl::VariadicFlag,
             std::string &) { return builder.getFunctionType(arg_types, results); };
      return function_interface_impl::parseFunctionOp(
          parser, result, /*allowVariadic=*/false,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  10. src/runtime/heapdump.go

    		// locals.
    		dumpbv(&bv, s.varp-uintptr(bv.n)*goarch.PtrSize-s.sp)
    	}
    	dumpint(fieldKindEol)
    
    	// Record arg info for parent.
    	child.argoff = s.argp - s.fp
    	child.arglen = s.argBytes()
    	child.sp = (*uint8)(unsafe.Pointer(s.sp))
    	child.depth++
    	stkmap = (*stackmap)(funcdata(f, abi.FUNCDATA_ArgsPointerMaps))
    	if stkmap != nil {
    		child.args = stackmapdata(stkmap, pcdata)
    	} else {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Tue Apr 09 04:07:57 UTC 2024
    - 17.6K bytes
    - Viewed (0)
Back to top