Searched refs:resize_function (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | softmax.cc | 161 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in Softmax()
209 desc->resize_function = [](const std::map<ValueId, BHWC>& buffers) { in Softmax1x1()
|
| D | reshape.cc | 160 desc->resize_function = [attr](const std::map<ValueId, BHWC>& buffers) { in Reshape()
219 desc->resize_function = [attr](const std::map<ValueId, BHWC>& buffers) { in Reshapex4()
|
| D | concat.cc | 157 desc->resize_function = [input_ids](const std::map<ValueId, BHWC>& buffers) { in ConcatZ()
247 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in ConcatX()
338 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in ConcatY()
|
| D | depthwise_conv.cc | 610 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConvolution()
660 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride1x1()
724 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in DepthWiseConv3x3Stride2()
|
| D | mean.cc | 122 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in Mean()
|
| D | max_unpooling.cc | 129 desc->resize_function = [input_id, input_indices_id, in MaxUnpooling()
|
| D | resize.cc | 145 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in Resize()
|
| D | add.cc | 128 desc->resize_function = [input_ids](const std::map<ValueId, BHWC>& buffers) { in Add()
|
| D | conv.cc | 953 desc->resize_function = [output_id, in Convolution()
1010 desc->resize_function = [output_id, in Convolution1x1()
1068 desc->resize_function = [output_id, in ConvolutionGeneric()
1122 desc->resize_function = [output_id, in ConvolutionPrecise()
1180 desc->resize_function = [output_id, in ConvolutionPrecise1x1PowerVR()
|
| D | mul.cc | 118 desc->resize_function = [input_id_0, in ApplyMask()
|
| D | elementwise.cc | 122 desc->resize_function = [input_ids](const std::map<ValueId, BHWC>& buffers) { in ElementwiseWithTwoInputs()
|
| D | padding.cc | 196 desc->resize_function = [input_id, in Padding()
|
| D | fully_connected.cc | 184 desc->resize_function = [attr](const std::map<ValueId, BHWC>& buffers) { in FullyConnected()
|
| D | pooling.cc | 242 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) { in PoolingInternal()
|
| D | slice.cc | 171 desc->resize_function = [input_id, in Slice()
|
| D | transpose_conv.cc | 979 desc->resize_function = [input_id, in ConvolutionTransposed()
1093 border_desc->resize_function = in ConvolutionTransposed3x3()
1166 desc->resize_function = [input_id, in ConvolutionTransposed3x3()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task_descriptor.h | 111 DispatchParamsFunction resize_function; member
|
| D | compiled_model.cc | 431 desc->resize_function = [input_id](const std::map<ValueId, BHWC>& buffers) { in NonLinkableStub()
549 fused_desciptor->resize_function = non_linkable->resize_function; in FuseChain()
|
| D | compiled_model_test.mm | 89 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) {
145 desc->resize_function = [input_id1](const std::map<ValueId, BHWC>& buffers) {
|
| D | compute_task.mm | 147 _resizeFunction = desc->resize_function;
|
| D | inference_context_test.mm | 96 desc->resize_function = [output_id](const std::map<ValueId, BHWC>& buffers) {
|