Home
last modified time | relevance | path

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

/external/mesa3d/src/gallium/drivers/r600/sfn/
Dsfn_instruction_tex.cpp238 int bias_idx = nir_tex_instr_src_index(tex, nir_tex_src_bias); in lower_txl_txf_array_or_cube() local
240 assert (lod_idx >= 0 || bias_idx >= 0); in lower_txl_txf_array_or_cube()
247 if (bias_idx >= 0) in lower_txl_txf_array_or_cube()
248 lod = nir_fadd(b, lod,nir_ssa_for_src(b, tex->src[bias_idx].src, 1)); in lower_txl_txf_array_or_cube()
272 if (bias_idx >= 0) in lower_txl_txf_array_or_cube()
273 nir_tex_instr_remove_src(tex, bias_idx); in lower_txl_txf_array_or_cube()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc1103 for (const int bias_idx : property.biases) { in QuantizeBiases() local
1104 if (bias_idx >= op->inputs.size() || in QuantizeBiases()
1105 op->inputs[bias_idx] == kTfLiteOptionalTensor) { in QuantizeBiases()
1110 TensorT* bias_tensor = subgraph->tensors[op->inputs[bias_idx]].get(); in QuantizeBiases()
1112 if (utils::HasBuffer(model, subgraph, op->inputs[bias_idx])) { in QuantizeBiases()
1117 bias_idx, op->inputs.size(), in QuantizeBiases()
1141 bias_tensor->name.c_str(), op->inputs[bias_idx], in QuantizeBiases()
1298 for (const int bias_idx : property.biases) { in EnsureBiasScaleCompatibility() local
1299 if (bias_idx >= op->inputs.size() || in EnsureBiasScaleCompatibility()
1300 op->inputs[bias_idx] == kTfLiteOptionalTensor) { in EnsureBiasScaleCompatibility()
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_lower_tex.c225 int bias_idx = nir_tex_instr_src_index(tex, nir_tex_src_bias); in lower_implicit_lod() local
226 if (bias_idx >= 0) { in lower_implicit_lod()
228 lod = nir_fadd(b, lod, nir_ssa_for_src(b, tex->src[bias_idx].src, 1)); in lower_implicit_lod()
229 nir_tex_instr_remove_src(tex, bias_idx); in lower_implicit_lod()
/external/tensorflow/tensorflow/lite/python/
Dlite_test.py1143 bias_idx = 1 if enable_mlir_converter else 0
1153 self.assertEqual(interpreter.get_tensor_details()[bias_idx]['name'],
1155 self.assertEqual(interpreter.get_tensor_details()[bias_idx]['dtype'],
1160 bias_idx = 2
1186 self.assertEqual(interpreter.get_tensor_details()[bias_idx]['name'],
1191 self.assertEqual(interpreter.get_tensor_details()[bias_idx]['dtype'],
1195 self.assertEqual(interpreter.get_tensor_details()[bias_idx]['dtype'],