Searched defs:idx_input (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLStackLayerKernel.cpp | 45 Status validate_arguments(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsi… in validate_arguments()
82 …ayerKernel::configure(const ICLTensor *input, unsigned int axis, unsigned int idx_input, unsigned … in configure()
87 …ext &compile_context, const ICLTensor *input, unsigned int axis, unsigned int idx_input, unsigned … in configure()
115 …yerKernel::validate(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned … in validate()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEStackLayerKernel.cpp | 44 Status validate_arguments(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsi… in validate_arguments()
74 …_from_axis_and_replace_coordinate(const Coordinates &id, unsigned int axis, unsigned int idx_input) in shift_from_axis_and_replace_coordinate()
92 void NEStackLayerKernel::configure(const ITensor *input, unsigned int axis, unsigned int idx_input,… in configure()
109 …yerKernel::validate(const ITensorInfo *input, unsigned int axis, unsigned int idx_input, unsigned … in validate()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_api_test.cpp | 4251 for (int idx_input = 1; (unsigned)idx_input < input_shape.size() + 1; ++idx_input) { in test_repeat() local
7051 for (int idx_input = 1; (unsigned)idx_input < input_shape.size() + 1; ++idx_input) { in test_tile() local
|