Home
last modified time | relevance | path

Searched refs:in_idx (Results 1 – 19 of 19) 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/kernels/image/
Dresize_bilinear_op_gpu.cu.cc175 GPU_1D_KERNEL_LOOP(in_idx, nthreads) { in ResizeBilinearGradKernel()
177 int idx = in_idx; in ResizeBilinearGradKernel()
202 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in ResizeBilinearGradKernel()
216 const float dbottom = y_lerp * input_grad[in_idx]; in ResizeBilinearGradKernel()
265 int in_idx = in ResizeBilinearDeterministicGradKernel() local
274 acc += static_cast<T>(input_grad[in_idx] * y_lerp * x_lerp); in ResizeBilinearDeterministicGradKernel()
341 GPU_1D_KERNEL_LOOP(in_idx, nthreads) { in LegacyResizeBilinearGradKernel()
343 int idx = in_idx; in LegacyResizeBilinearGradKernel()
365 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in LegacyResizeBilinearGradKernel()
379 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.c99 unsigned in_idx; in u_index_translator() local
109 in_idx = in_size_idx(in_index_size); in u_index_translator()
129 *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.h138 int in_idx = j * stride + k; in RunCoeffCheck() local
140 input_block[in_idx] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_); in RunCoeffCheck()
190 int in_idx = j * pitch_ + k; in RunInvCoeffCheck() local
192 input_block[in_idx] = (rnd.Rand16() & mask_) - (rnd.Rand16() & mask_); in RunInvCoeffCheck()
/external/deqp-deps/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.h130 void MarkOperandTypeAsFullyUsed(const Instruction* inst, uint32_t in_idx);
Deliminate_dead_members_pass.cpp249 const Instruction* inst, uint32_t in_idx) { in MarkOperandTypeAsFullyUsed() argument
250 uint32_t op_id = inst->GetSingleWordInOperand(in_idx); in MarkOperandTypeAsFullyUsed()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Deliminate_dead_members_pass.h130 void MarkOperandTypeAsFullyUsed(const Instruction* inst, uint32_t in_idx);
Deliminate_dead_members_pass.cpp249 const Instruction* inst, uint32_t in_idx) { in MarkOperandTypeAsFullyUsed() argument
250 uint32_t op_id = inst->GetSingleWordInOperand(in_idx); in MarkOperandTypeAsFullyUsed()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Deliminate_dead_members_pass.h130 void MarkOperandTypeAsFullyUsed(const Instruction* inst, uint32_t in_idx);
Deliminate_dead_members_pass.cpp249 const Instruction* inst, uint32_t in_idx) { in MarkOperandTypeAsFullyUsed() argument
250 uint32_t op_id = inst->GetSingleWordInOperand(in_idx); in MarkOperandTypeAsFullyUsed()
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils.py381 in_idx = np.ravel_multi_index(
384 yield (out_idx, in_idx)
/external/adhd/cras/src/server/
Dcras_fmt_conv_ops.c489 unsigned in_idx = 0; in s16_convert_channels() local
497 ch_conv_mtx[i], &in[in_idx], num_in_ch); in s16_convert_channels()
498 in_idx += num_in_ch; in s16_convert_channels()
/external/tensorflow/tensorflow/core/kernels/
Ddequantize_op_test.cc144 int in_idx = rng() % data.size(); in ScalePerSliceAlongAxis() local
146 out[out_idx] = data[in_idx] * multiplier; in ScalePerSliceAlongAxis()
Dquantize_op_test.cc78 int in_idx = rng() % data.size(); in ScalePerSliceAlongAxis() local
80 out[out_idx] = data[in_idx] * multiplier; in ScalePerSliceAlongAxis()
Dquantize_and_dequantize_op_test.cc114 int in_idx = rand_r(&seed) % data.size(); in ScalePerSliceAlongAxis() local
116 out[out_idx] = data[in_idx] * multiplier; in ScalePerSliceAlongAxis()
/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_tgsi.c707 uint in_idx = inst->Src[0].Register.Index; in etna_compile_pass_optimize_outputs() local
715 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].last_use == inst_idx) { in etna_compile_pass_optimize_outputs()
717 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].native; in etna_compile_pass_optimize_outputs()
719 c->file[TGSI_FILE_TEMPORARY].reg[in_idx].last_use = ETNA_MAX_TOKENS; in etna_compile_pass_optimize_outputs()
732 !c->file[TGSI_FILE_INPUT].reg[in_idx].native.valid && in etna_compile_pass_optimize_outputs()
737 c->file[TGSI_FILE_INPUT].reg[in_idx].native = in etna_compile_pass_optimize_outputs()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc170 int64 in_idx = data[i]; in TransposeShapeFn() local
171 if (in_idx >= rank) { in TransposeShapeFn()
172 return errors::InvalidArgument("perm dim ", in_idx, in TransposeShapeFn()
175 dims[i] = c->Dim(input, in_idx); in TransposeShapeFn()