Home
last modified time | relevance | path

Searched refs:target_index (Results 1 – 20 of 20) 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.cc235 [&, this](const ShapeIndex& target_index, in HandleGetTupleElement()
241 for (auto element : target_index) { in HandleGetTupleElement()
248 points_to_set.add_tuple_source(target_index, tuple); in HandleGetTupleElement()
329 ShapeIndex target_index({0}); in HandleSend() local
331 target_index.push_back(element); in HandleSend()
333 *points_to_set.mutable_element(target_index) = points_to; in HandleSend()
337 points_to_set.add_tuple_source(target_index, tuple); in HandleSend()
363 ShapeIndex target_index; in HandleTuple() local
364 target_index.push_back(i); in HandleTuple()
366 target_index.push_back(element); in HandleTuple()
[all …]
Delemental_ir_emitter.cc1086 const llvm_ir::IrArray::Index& target_index, const HloInstruction& hlo, in ElementwiseSourceIndex() argument
1100 return target_index; in ElementwiseSourceIndex()
1109 source_index.push_back(target_index[i]); in ElementwiseSourceIndex()
1404 const IrArray::Index target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
1406 auto source_index = target_index; in MakeElementGenerator()
1472 const IrArray::Index& target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
1474 auto source_index = target_index; in MakeElementGenerator()
1477 llvm::ConstantInt::get(target_index[dim]->getType(), in MakeElementGenerator()
1479 target_index[dim]); in MakeElementGenerator()
1485 const IrArray::Index& target_index) -> StatusOr<llvm::Value*> { in MakeElementGenerator() argument
[all …]
Delemental_ir_emitter.h137 const llvm_ir::IrArray::Index& target_index, const HloInstruction& hlo,
Dhlo_evaluator.cc1190 std::vector<int64> target_index(ShapeUtil::Rank(result->shape()), 0); in HandlePad() local
1201 target_index[i] = in HandlePad()
1207 if (!(target_index[i] >= 0 && in HandlePad()
1208 target_index[i] < pad->shape().dimensions(i))) { in HandlePad()
1212 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/libtextclassifier/
Dtextclassifier_jni.cc200 const int* target_index; in ConvertIndicesBMPUTF8() local
203 target_index = &bmp_index; in ConvertIndicesBMPUTF8()
206 target_index = &unicode_index; in ConvertIndicesBMPUTF8()
211 &source_index, &target_index]() { in ConvertIndicesBMPUTF8()
213 result.first = *target_index; in ConvertIndicesBMPUTF8()
217 result.second = *target_index; in ConvertIndicesBMPUTF8()
/external/mesa3d/src/compiler/glsl/
Dlinker.cpp2529 unsigned target_index) in assign_attribute_or_color_locations() argument
2535 unsigned max_index = (target_index == MESA_SHADER_VERTEX) ? in assign_attribute_or_color_locations()
2536 constants->Program[target_index].MaxAttribs : in assign_attribute_or_color_locations()
2545 assert((target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2546 || (target_index == MESA_SHADER_FRAGMENT)); in assign_attribute_or_color_locations()
2548 gl_linked_shader *const sh = prog->_LinkedShaders[target_index]; in assign_attribute_or_color_locations()
2568 const int generic_base = (target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2572 (target_index == MESA_SHADER_VERTEX) in assign_attribute_or_color_locations()
2619 } else if (target_index == MESA_SHADER_VERTEX) { in assign_attribute_or_color_locations()
2627 } else if (target_index == MESA_SHADER_FRAGMENT) { in assign_attribute_or_color_locations()
[all …]
/external/mesa3d/src/mesa/main/
Dtexstate.c514 gl_texture_index target_index; in update_single_program_texture() local
536 target_index = ffs(prog->TexturesUsed[unit]) - 1; in update_single_program_texture()
537 texObj = texUnit->CurrentTex[target_index]; in update_single_program_texture()
563 texObj = _mesa_get_fallback_texture(ctx, target_index); in update_single_program_texture()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc575 llvm_ir::IrArray::Index target_index; in HandleDot() local
578 target_index.push_back(lhs_index[dimension]); in HandleDot()
583 target_index.push_back(rhs_index[dimension]); in HandleDot()
588 target_index, in HandleDot()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc124 const auto& target_index = *this; in SourceIndexOfReshape() local
125 CHECK_EQ(target_index.size(), ShapeUtil::Rank(output_shape)); in SourceIndexOfReshape()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc856 llvm_ir::IrArray::Index target_index; in Emit() local
859 target_index.push_back(lhs_index[dimension]); in Emit()
864 target_index.push_back(rhs_index[dimension]); in Emit()
868 target_array_.EmitWriteArrayElement(target_index, result, ir_builder_); in Emit()
Dir_emitter.cc660 [this, init_value](const llvm_ir::IrArray::Index& target_index) { in HandleSelectAndScatter() argument
1915 llvm_ir::IrArray::Index target_index = in HandleSlice() local
1921 std::replace(target_index.begin(), target_index.end(), in HandleSlice()
1930 const llvm_ir::IrArray::Index source_index = target_index.SourceIndexOfSlice( in HandleSlice()
1935 target_index, &ir_builder_, "slice.dest"); in HandleSlice()
2011 [this, pad](const llvm_ir::IrArray::Index& target_index) { in HandlePad() argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster.cc264 void WhenFound(int target_index, const Status& s, in WhenFound() argument
267 seen_targets_[target_index] = true; in WhenFound()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_ratectrl.c1800 int target_index = rc->worst_quality; in vp9_compute_qdelta() local
1811 target_index = i; in vp9_compute_qdelta()
1815 return target_index - start_index; in vp9_compute_qdelta()
1821 int target_index = rc->worst_quality; in vp9_compute_qdelta_by_rate() local
1835 target_index = i; in vp9_compute_qdelta_by_rate()
1839 return target_index - qindex; in vp9_compute_qdelta_by_rate()
/external/v8/src/interpreter/
Dinterpreter-assembler.h255 void TraceBytecodeDispatch(compiler::Node* target_index);
Dinterpreter-assembler.cc1143 Node* target_index = IntPtrAdd(base_index, next_bytecode); in DispatchWide() local
1146 WordShl(target_index, kPointerSizeLog2)); in DispatchWide()
/external/v8/src/s390/
Dassembler-s390.cc1780 int32_t target_index = emit_code_target(target, rmode, ast_id); in call() local
1781 brasl(r14, Operand(target_index)); in call()
1788 int32_t target_index = emit_code_target(target, rmode); in jump() local
1789 brcl(cond, Operand(target_index)); in jump()
/external/v8/src/
Dobjects.cc3456 int target_index = details.field_index() - new_map->GetInObjectProperties(); in MigrateFastToFast() local
3457 DCHECK(target_index >= 0); // Must be a backing store index. in MigrateFastToFast()
3458 new_storage->set(target_index, *value); in MigrateFastToFast()
3540 int target_index = new_descriptors->GetFieldIndex(i) - inobject; in MigrateFastToFast() local
3541 if (target_index < 0) target_index += total_size; in MigrateFastToFast()
3542 array->set(target_index, *value); in MigrateFastToFast()
3555 int target_index = new_descriptors->GetFieldIndex(i) - inobject; in MigrateFastToFast() local
3556 if (target_index < 0) target_index += total_size; in MigrateFastToFast()
3557 array->set(target_index, *value); in MigrateFastToFast()