Home
last modified time | relevance | path

Searched refs:src_def (Results 1 – 5 of 5) 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
217 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
220 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
223 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
242 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis, gpu_info) && in GenerateConvolutionTransposedCode()
277 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
300 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
333 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
347 if (!src_def.SupportsZeroClamp(Axis::DEPTH, gpu_info)) { in GenerateConvolutionTransposedCode()
350 if (!src_def.CanReadOutOfBorder(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
[all …]
Dconv_generic.cc379 const auto& src_def = op_def.src_tensors[0]; in GenerateConv() local
384 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConv()
387 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConv()
390 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
412 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis, gpu_info) && in GenerateConv()
435 if (src_def.HasAxis(Axis::DEPTH) && !conv_params_.z_kernel_is_1) { in GenerateConv()
499 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
527 src_def.HasAxis(Axis::DEPTH), in GenerateConv()
528 src_def.HasAxis(Axis::BATCH)); in GenerateConv()
529 if (src_def.HasAxis(Axis::BATCH)) { 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/libbpf/src/
Dlinker.c1576 struct btf_map_def src_def = {}, src_inner_def = {}; in glob_map_defs_match() local
1587 err = parse_btf_map_def(sym_name, obj->btf, t, true /*strict*/, &src_def, &src_inner_def); in glob_map_defs_match()
1608 obj->btf, &src_def, &src_inner_def); in glob_map_defs_match()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc977 const NodeDef& src_def = src_node->def(); in IsTensorRTCandidate() local
980 Status status = ConvertToTensorOrWeights(src_def, edge->src_output(), in IsTensorRTCandidate()
983 VLOG(2) << "Failed to convert input `" << src_def.name() << "` to a " in IsTensorRTCandidate()