Home
last modified time | relevance | path

Searched refs:in_idx (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsqueeze.cc72 for (int in_idx = 0, out_idx = 0; in_idx < input_num_dims; ++in_idx) { in Prepare() local
73 if (!should_squeeze[in_idx]) { in Prepare()
74 output_dims->data[out_idx++] = input_dims->data[in_idx]; in Prepare()
/external/tensorflow/tensorflow/core/ops/
Dtpu_cross_replica_ops.cc61 int64 in_idx = i; in __anonb72bce3d0102() local
63 in_idx = split_dimension; in __anonb72bce3d0102()
65 in_idx = concat_dimension; in __anonb72bce3d0102()
68 dims[i] = c->Dim(input, in_idx); in __anonb72bce3d0102()
Darray_ops.cc163 int64 in_idx = data[i]; in TransposeShapeFn() local
164 if (in_idx >= rank) { in TransposeShapeFn()
165 return errors::InvalidArgument("perm dim ", in_idx, in TransposeShapeFn()
168 dims[i] = c->Dim(input, in_idx); in TransposeShapeFn()
/external/tensorflow/tensorflow/core/kernels/
Dresize_bilinear_op_gpu.cu.cc92 CUDA_1D_KERNEL_LOOP(in_idx, nthreads) { in ResizeBilinearGradKernel()
94 int idx = in_idx; in ResizeBilinearGradKernel()
119 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in ResizeBilinearGradKernel()
133 const float dbottom = y_lerp * input_grad[in_idx]; in ResizeBilinearGradKernel()
206 CUDA_1D_KERNEL_LOOP(in_idx, nthreads) { in LegacyResizeBilinearGradKernel()
208 int idx = in_idx; in LegacyResizeBilinearGradKernel()
230 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in LegacyResizeBilinearGradKernel()
244 const float dbottom = y_lerp * input_grad[in_idx]; in LegacyResizeBilinearGradKernel()
/external/mesa3d/src/gallium/auxiliary/indices/
Du_unfilled_indices.c138 unsigned in_idx; in u_unfilled_translator() local
145 in_idx = in_size_idx(in_index_size); in u_unfilled_translator()
173 *out_translate = translate_line[in_idx][out_idx][prim]; in u_unfilled_translator()
Du_indices.c86 unsigned in_idx; in u_index_translator() local
96 in_idx = in_size_idx(in_index_size); in u_index_translator()
116 *out_translate = translate[in_idx][out_idx][in_pv][out_pv][prim_restart][prim]; in u_index_translator()
/external/libaom/libaom/test/
Dtransform_test_base.h135 int in_idx = j * stride + k; in RunCoeffCheck() local
137 input_block[in_idx] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_); in RunCoeffCheck()
187 int in_idx = j * pitch_ + k; in RunInvCoeffCheck() local
189 input_block[in_idx] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_); in RunInvCoeffCheck()
/external/libxcam/xcore/interface/
Dstitcher.h114 uint32_t in_idx; member
119 : in_idx (INVALID_INDEX) in CopyArea()
Dstitcher.cpp40 if (current.in_idx == next.in_idx && in merge_neighbor_area()
453 left.in_idx = i; in update_copy_areas()
474 right.in_idx = next_i; in update_copy_areas()
/external/libxcam/modules/soft/
Dsoft_stitcher.cpp315 area.in_idx != INVALID_INDEX && in create_copier()
318 "stitcher: copy area (idx:%d) is invalid", area.in_idx); in create_copier()
374 areas[i].in_idx, in init_config()
378 XCAM_ASSERT (areas[i].in_idx < size); in init_config()
671 if(_copiers[i].copy_area.in_idx == idx) { in start_copy_tasks()
/external/adhd/cras/src/server/
Dcras_fmt_conv.c433 unsigned in_idx = 0; in convert_channels() local
440 &in[in_idx], in convert_channels()
442 in_idx += conv->in_fmt.num_channels; in convert_channels()
/external/mesa3d/src/gallium/auxiliary/tgsi/
Dtgsi_lowering.c1180 unsigned in_idx = ctx->two_side_idx[i]; in emit_twoside() local
1186 decl.Semantic.Index = info->input_semantic_index[in_idx]; in emit_twoside()
1188 decl.Interp.Interpolate = info->input_interpolate[in_idx]; in emit_twoside()
1189 decl.Interp.Location = info->input_interpolate_loc[in_idx]; in emit_twoside()
1190 decl.Interp.CylindricalWrap = info->input_cylindrical_wrap[in_idx]; in emit_twoside()
/external/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_compiler.c693 uint in_idx = inst->Src[0].Register.Index; in etna_compile_pass_optimize_outputs() local
701 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].last_use == inst_idx) { in etna_compile_pass_optimize_outputs()
703 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].native; in etna_compile_pass_optimize_outputs()
705 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].last_use = ETNA_MAX_TOKENS; in etna_compile_pass_optimize_outputs()
718 !c->file[TGSI_FILE_INPUT].reg[in_idx].native.valid && in etna_compile_pass_optimize_outputs()
723 c->file[TGSI_FILE_INPUT].reg[in_idx].native = in etna_compile_pass_optimize_outputs()