Searched defs:tuple_arg (Results 1 – 7 of 7) sorted by relevance
| /third_party/python/Lib/lib2to3/fixes/ |
| D | fix_tuple_params.py | 67 def handle_tuple(tuple_arg, add_prefix=False): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/construct_input/ |
| D | test_outermost_input_pijit.py | 45 tuple_arg = (tensor_x, tensor_y, tensor_z, tensor_w) variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 728 bool IsTupleAllTensor(const AbstractTuplePtr &tuple_arg) { in IsTupleAllTensor()
739 bool EnableGradFirstForTuple(const AbstractTuplePtr &tuple_arg, bool enable_tuple_grad) { in EnableGradFirstForTuple()
749 bool CanGradArgument(const AbstractTuplePtr &tuple_arg, size_t pos) { in CanGradArgument()
755 void GenerateFuncGraphByPosition(const FuncGraphPtr &fg, const AbstractTuplePtr &tuple_arg, const A… in GenerateFuncGraphByPosition()
845 FuncGraphPtr Tail::GenerateGradFuncGraph(const AbstractTuplePtr &tuple_arg, const AbstractTuplePtr … in GenerateGradFuncGraph() argument
916 AbstractTuplePtr tuple_arg = dyn_cast<AbstractTuple>(args_abs_list[0]); in GenerateFuncGraph() local
952 AbstractTuplePtr tuple_arg = dyn_cast<AbstractTuple>(args_abs_list[0]); in GenerateFuncGraph() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/pipeline/parse/ |
| D | test_outermost_net_pass_non_tensor_inputs.py | 68 tuple_arg = (tensor_x, tensor_y, tensor_z, tensor_w) variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | strided_slice_v2.cc | 366 auto tuple_arg = input_arg->GetShape()->cast<abstract::SequenceShapePtr>(); in CheckAndGetDynamicSliceV2() local
|
| /third_party/rust/rust/compiler/rustc_mir_transform/src/ |
| D | shim.rs | 783 let tuple_arg = Local::new(1 + (sig.inputs().len() - 1)); in build_call_shim() localVariable
|
| /third_party/mindspore/mindspore-src/source/tests/st/construct_input/ |
| D | test_outermost_input.py | 67 tuple_arg = (tensor_x, tensor_y, tensor_z, tensor_w) variable
|