Home
last modified time | relevance | path

Searched refs:target_index (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dinput_target.h31 int target_index) const = 0;
36 int target_index) const = 0;
64 int target_index) const override { in GetTargetAsClassIndex() argument
66 GetTargetAsContinuous(example_index, target_index)); in GetTargetAsClassIndex()
77 int target_index) const override { in GetTargetAsContinuous() argument
78 QCHECK_LT(target_index, num_targets_); in GetTargetAsContinuous()
79 return (*target_)(example_index * num_targets_ + target_index); in GetTargetAsContinuous()
Dtest_utils.h33 int target_index) const override { in GetTargetAsClassIndex() argument
35 GetTargetAsContinuous(example_index, target_index)); in GetTargetAsClassIndex()
46 int target_index) const override { in GetTargetAsContinuous() argument
47 QCHECK_LT(target_index, num_targets_); in GetTargetAsContinuous()
48 return (*target_)[example_index * num_targets_ + target_index]; in GetTargetAsContinuous()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc232 [&](const ShapeIndex& target_index, PointsToSet::BufferList* points_to) { in HandleGetTupleElement() argument
237 for (auto element : target_index) { in HandleGetTupleElement()
244 points_to_set.add_tuple_source(target_index, tuple); in HandleGetTupleElement()
341 ShapeIndex target_index({0}); in HandleSend() local
343 target_index.push_back(element); in HandleSend()
345 *points_to_set.mutable_element(target_index) = points_to; in HandleSend()
349 points_to_set.add_tuple_source(target_index, tuple); in HandleSend()
375 ShapeIndex target_index; in HandleTuple() local
376 target_index.push_back(i); in HandleTuple()
378 target_index.push_back(element); in HandleTuple()
[all …]
Delemental_ir_emitter.cc1708 const llvm_ir::IrArray::Index& target_index) { in EmitElementalConcatenate() argument
1710 auto source_index = target_index; in EmitElementalConcatenate()
2280 const IrArray::Index target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
2282 target_index); in MakeElementGenerator()
2286 const IrArray::Index& target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
2288 std::vector<llvm::Value*> source_multi_index = target_index.multidim(); in MakeElementGenerator()
2290 source_multi_index[dim] = Sub(target_index.GetConstantWithIndexType( in MakeElementGenerator()
2292 target_index[dim]); in MakeElementGenerator()
2295 source_multi_index, operand->shape(), target_index.GetType()); in MakeElementGenerator()
2300 const IrArray::Index& target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
[all …]
Delemental_ir_emitter.h178 const llvm_ir::IrArray::Index& target_index);
Dhlo_evaluator_typed_visitor.h1417 std::vector<int64> target_index(result.shape().rank(), 0); in HandlePad()
1428 target_index[i] = in HandlePad()
1434 if (!(target_index[i] >= 0 && in HandlePad()
1435 target_index[i] < pad->shape().dimensions(i))) { in HandlePad()
1439 result.Set<ReturnT>(target_index, in HandlePad()
/external/libxcam/cl_kernel/
Dkernel_tonemapping.cl50 …int target_index = local_index <= SHARED_PIXEL_X_SIZE ? local_index : (local_index <= (SHARED_PIXE…
53 int offset_x = target_index % SHARED_PIXEL_X_SIZE;
54 int offset_y = target_index / SHARED_PIXEL_X_SIZE;
67 local_src_data[target_index] = y_data;
/external/v8/src/s390/
Dassembler-s390.cc674 int32_t target_index = AddCodeTarget(target); in call() local
675 brasl(r14, Operand(target_index)); in call()
682 int32_t target_index = AddCodeTarget(Handle<Code>()); in call() local
683 brasl(r14, Operand(target_index)); in call()
692 int32_t target_index = AddCodeTarget(target); in jump() local
693 brcl(cond, Operand(target_index)); in jump()
/external/libtextclassifier/annotator/
Dannotator_jni.cc310 const int* target_index; in ConvertIndicesBMPUTF8() local
313 target_index = &bmp_index; in ConvertIndicesBMPUTF8()
316 target_index = &unicode_index; in ConvertIndicesBMPUTF8()
321 &source_index, &target_index]() { in ConvertIndicesBMPUTF8()
323 result.first = *target_index; in ConvertIndicesBMPUTF8()
327 result.second = *target_index; in ConvertIndicesBMPUTF8()
/external/mesa3d/src/mesa/main/
Dtexstate.c643 gl_texture_index target_index; in update_single_program_texture() local
660 target_index = ffs(prog->TexturesUsed[unit]) - 1; in update_single_program_texture()
661 texObj = texUnit->CurrentTex[target_index]; in update_single_program_texture()
687 texObj = _mesa_get_fallback_texture(ctx, target_index); in update_single_program_texture()
845 const gl_texture_index target_index = ffs(prog->TexturesUsed[unit]) - 1; in fix_missing_textures_for_atifs() local
849 _mesa_get_fallback_texture(ctx, target_index); in fix_missing_textures_for_atifs()
/external/mesa3d/src/compiler/glsl/
Dlinker.cpp2600 unsigned target_index) in assign_attribute_or_color_locations() argument
2606 unsigned max_index = (target_index == MESA_SHADER_VERTEX) ? in assign_attribute_or_color_locations()
2607 constants->Program[target_index].MaxAttribs : in assign_attribute_or_color_locations()
2615 assert((target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2616 || (target_index == MESA_SHADER_FRAGMENT)); in assign_attribute_or_color_locations()
2618 gl_linked_shader *const sh = prog->_LinkedShaders[target_index]; in assign_attribute_or_color_locations()
2638 const int generic_base = (target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2642 (target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2691 } else if (target_index == MESA_SHADER_VERTEX) { in assign_attribute_or_color_locations()
2699 } else if (target_index == MESA_SHADER_FRAGMENT) { in assign_attribute_or_color_locations()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster.cc311 void WhenFound(int target_index, const Status& s, in WhenFound() argument
314 seen_targets_[target_index] = true; in WhenFound()
317 << targets_[target_index] << " returned error: " << s; in WhenFound()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc133 const auto& target_index = *this; in SourceIndexOfReshape() local
134 CHECK_EQ(target_index.size(), output_shape.rank()); in SourceIndexOfReshape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc623 llvm_ir::IrArray::Index target_index(target_multi_index, in HandleDot() local
626 target_index, in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc785 [this, init_value](const llvm_ir::IrArray::Index& target_index) { in HandleSelectAndScatter() argument
2017 llvm_ir::IrArray::Index target_index(target_multi_index, slice->shape(), in HandleSlice() local
2025 const llvm_ir::IrArray::Index source_index = target_index.SourceIndexOfSlice( in HandleSlice()
2030 target_array.EmitArrayElementAddress(target_index, &b_, "slice.dest"); in HandleSlice()
2106 [this, pad](const llvm_ir::IrArray::Index& target_index) { in HandlePad() argument
2425 llvm_ir::IrArray::Index target_index(target_multi_index, output_shape, in EmitFastConcatenate() local
2439 target_array.EmitArrayElementAddress(target_index, &b_, "target_region"), in EmitFastConcatenate()
2455 source_array.EmitArrayElementAddress(target_index, &b_, "src_addr"), in EmitFastConcatenate()
Ddot_op_emitter.cc537 llvm_ir::IrArray::Index target_index( in EmitNaiveLlvmIrGemm() local
539 target_array_.EmitWriteArrayElement(target_index, result, b_); in EmitNaiveLlvmIrGemm()
/external/libaom/libaom/av1/encoder/
Dratectrl.c1717 const int target_index = in av1_compute_qdelta() local
1719 return target_index - start_index; in av1_compute_qdelta()
1760 const int target_index = in av1_compute_qdelta_by_rate() local
1763 return target_index - qindex; in av1_compute_qdelta_by_rate()
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_memory.cpp650 const auto target_index = 0; in ValidateCopyMemory() local
651 const auto target_id = inst->GetOperandAs<uint32_t>(target_index); in ValidateCopyMemory()
/external/v8/src/interpreter/
Dinterpreter-assembler.h305 void TraceBytecodeDispatch(compiler::Node* target_index);
Dinterpreter-assembler.cc1460 Node* target_index = IntPtrAdd(base_index, next_bytecode); in DispatchWide() local
1463 TimesPointerSize(target_index)); in DispatchWide()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_ratectrl.c2312 int target_index = rc->worst_quality; in vp9_compute_qdelta() local
2323 target_index = i; in vp9_compute_qdelta()
2327 return target_index - start_index; in vp9_compute_qdelta()
2333 int target_index = rc->worst_quality; in vp9_compute_qdelta_by_rate() local
2347 target_index = i; in vp9_compute_qdelta_by_rate()
2351 return target_index - qindex; in vp9_compute_qdelta_by_rate()
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_memory.cpp822 const auto target_index = 0; in ValidateCopyMemory() local
823 const auto target_id = inst->GetOperandAs<uint32_t>(target_index); in ValidateCopyMemory()
/external/v8/src/builtins/
Dbuiltins-array-gen.cc3734 Node* const target_index = var_target_index.value(); in FlattenIntoArray() local
3791 element_length, target_index, NumberDec(depth))); in FlattenIntoArray()
3808 element_length, target_index, NumberDec(depth))); in FlattenIntoArray()
3817 target_index, NumberConstant(kMaxSafeInteger), &throw_error); in FlattenIntoArray()
3823 target_index, element); in FlattenIntoArray()
3826 var_target_index.Bind(NumberInc(target_index)); in FlattenIntoArray()
3831 source_length, target_index); in FlattenIntoArray()
/external/v8/src/
Dobjects.cc4141 int target_index = details.field_index() - new_map->GetInObjectProperties(); in MigrateFastToFast() local
4144 (details.location() == kField && target_index >= 0 && in MigrateFastToFast()
4145 property_array_length > target_index); in MigrateFastToFast()
4150 (have_space && ((FLAG_unbox_double_fields && target_index < 0) || in MigrateFastToFast()
4185 DCHECK_GE(target_index, 0); // Must be a backing store index. in MigrateFastToFast()
4186 new_storage->set(target_index, *value); in MigrateFastToFast()
4275 int target_index = new_descriptors->GetFieldIndex(i); in MigrateFastToFast() local
4276 if (target_index < inobject) { in MigrateFastToFast()
4277 inobject_props->set(target_index, *value); in MigrateFastToFast()
4279 array->set(target_index - inobject, *value); in MigrateFastToFast()
[all …]