Searched refs:target_element (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 50 TF_ASSIGN_OR_RETURN(llvm::Value * target_element, in LoopEmitter()
52 target_array.EmitWriteArrayElement(array_index, target_element, b); in LoopEmitter()
62 TF_ASSIGN_OR_RETURN(llvm::Value * target_element, in MakeBodyEmitterForMultiOutput()
64 CHECK(target_element->getType()->isStructTy()) in MakeBodyEmitterForMultiOutput()
67 CHECK_EQ(target_element->getType()->getStructNumElements(), in MakeBodyEmitterForMultiOutput()
72 array_index, b->CreateExtractValue(target_element, i), b); in MakeBodyEmitterForMultiOutput()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/ |
| D | optimized_encoder.cc | 183 LatticeElement& target_element = lattice[i + m.match_length]; in EncodeNormalizedString() local
185 if (target_element.prev_position < 0 || target_element.score < score) { in EncodeNormalizedString()
186 target_element = LatticeElement(score, m.id, i); in EncodeNormalizedString()
|