Home
last modified time | relevance | path

Searched refs:src_def (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed.cc154 const auto& src_def = op_def.src_tensors[0]; in GenerateConvolutionTransposedCode() local
200 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
203 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
206 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
225 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConvolutionTransposedCode()
260 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
283 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
316 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
330 if (!src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
333 if (!src_def.CanReadOutOfBorder(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
[all …]
Dconv_powervr.cc360 const auto& src_def = op_def.src_tensors[0]; in GenerateConv() local
365 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConv()
368 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConv()
371 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
393 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConv()
422 if (src_def.HasAxis(Axis::DEPTH) && !conv_params_.z_kernel_is_1) { in GenerateConv()
485 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
513 src_def.HasAxis(Axis::DEPTH)); in GenerateConv()
563 if (!src_def.CanReadOutOfBorder(Axis::WIDTH)) { in GenerateConv()
580 if (!src_def.CanReadOutOfBorder(Axis::HEIGHT)) { in GenerateConv()
[all …]
/external/mesa3d/src/broadcom/compiler/
Dvir_opt_small_immediates.c62 struct qinst *src_def = c->defs[inst->src[i].index]; in vir_opt_small_immediates() local
63 if (!src_def || !src_def->qpu.sig.ldunif) in vir_opt_small_immediates()
65 int uniform = src_def->uniform; in vir_opt_small_immediates()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1210 const NodeDef& src_def = edge->src()->def(); in IsTensorRTCandidate() local
1211 Status status = ConvertToTensorOrWeights(src_def, edge->src_output(), in IsTensorRTCandidate()
1215 "Failed to convert input ", src_def.name(), in IsTensorRTCandidate()