Searched defs:output_args (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 50 const std::vector<ArgSpec>& output_args, in PopulateFunction()
75 const std::vector<ArgSpec> output_args{{"out", "float32"}}; in PopulateSampleLibrary() local
91 const std::vector<ArgSpec> output_args{{"out", "float32"}}; in PopulateComplexLibrary() local
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | reduction.h | 111 const std::vector<ExprPtr>& output_args, in complete()
122 const std::vector<ExprHandle>& output_args, in complete()
|
| D | tensor.cpp | 26 std::vector<ExprPtr> output_args(args.begin(), args.end()); in constructStmt() local
|
| /external/googleapis/google/cloud/visionai/v1/ |
| D | lva.proto | 89 repeated ArgumentDefinition output_args = 3; field
|
| /external/pytorch/torch/csrc/jit/mobile/nnc/ |
| D | context.cpp | 213 auto output_args = output_specs_.size(); in init_execution_state() local
|
| /external/openscreen/third_party/abseil/src/absl/flags/ |
| D | parse.cc | 627 std::vector<char*> output_args; in ParseCommandLineImpl() local
|
| /external/libkmsxx/utils/ |
| D | kmstest.cpp | 530 …<OutputInfo> setups_to_outputs(Card& card, ResourceManager& resman, const vector<Arg>& output_args) in setups_to_outputs()
1123 vector<Arg> output_args = parse_cmdline(argc, argv); in main() local
|
| /external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/flags/ |
| D | parse.cc | 685 std::vector<char*> output_args; in ParseCommandLineImpl() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 43 std::vector<OutputArgInstantiation> output_args, in GrapplerFunctionItem()
|
| /external/pytorch/torch/fx/ |
| D | graph.py | 791 def generate_output(self, output_args): argument
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_wrapper_cpu.py | 1989 self, op_overload, raw_args, output_args argument
|
| /external/pytorch/test/ |
| D | test_fx.py | 3816 def generate_output(self, output_args): argument
|