Searched refs:tmp_output (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEReduceMean.cpp | 128 ITensor *tmp_output = output; in configure() local
139 tmp_output = &_output_no_quant; in configure()
156 … _reduction_kernels[i].configure(in, tmp_output, axis_local[i], ReductionOperation::MEAN_SUM); in configure()
183 … auto_init_if_empty(*tmp_output->info(), tmp_input->info()->clone()->set_tensor_shape(out_shape)); in configure()
184 _reshape.configure(&_reduced_outs[_reduction_ops - 1], tmp_output); in configure()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLReduceMean.cpp | 127 ICLTensor *tmp_output = output; in configure() local
138 tmp_output = &_output_no_quant; in configure()
155 …_reduction_kernels[i].configure(compile_context, in, tmp_output, axis_local[i], ReductionOperation… in configure()
183 … auto_init_if_empty(*tmp_output->info(), tmp_input->info()->clone()->set_tensor_shape(out_shape)); in configure()
184 _reshape.configure(compile_context, &_reduced_outs[_reduction_ops - 1], tmp_output); in configure()
|
| D | CLSoftmaxLayer.cpp | 75 ICLTensor *tmp_output = output; in configure() local
82 tmp_output = &_output_permuted; in configure()
109 _norm_kernel->configure(compile_context, &_tmp, &_sum, tmp_output, softmax_info); in configure()
|
| /external/mesa3d/src/gallium/auxiliary/draw/ |
| D | draw_gs.h | 64 float (*tmp_output)[4]; member
|
| D | draw_gs.c | 450 &shader->stream[i].tmp_output); in gs_flush()
639 shader->stream[i].tmp_output = (float (*)[4])output_verts[i].verts->data; in draw_geometry_shader_run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2066 auto tmp_output = in HandleReshape() local
2088 auto reshard_output = tmp_output.ReshardAsWindowedInput( in HandleReshape()
|