| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 384 void ConvertConvOperator(const Model& model, const ConvOperator& src_op, in ConvertConvOperator() argument
386 const bool has_bias = src_op.inputs.size() >= 3; in ConvertConvOperator()
387 string conv_output = src_op.outputs[0]; in ConvertConvOperator()
395 *conv2d_op->add_input() = src_op.inputs[0]; in ConvertConvOperator()
396 *conv2d_op->add_input() = src_op.inputs[1]; in ConvertConvOperator()
399 WalkUpToConstantArray(model, src_op.inputs[1]); in ConvertConvOperator()
406 strides.mutable_list()->add_i(src_op.stride_height); in ConvertConvOperator()
407 strides.mutable_list()->add_i(src_op.stride_width); in ConvertConvOperator()
409 if ((src_op.dilation_width_factor != 1) || in ConvertConvOperator()
410 (src_op.dilation_height_factor != 1)) { in ConvertConvOperator()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 34 auto src_op = op_it->get(); in Run() local
35 if (src_op->type != OperatorType::kLstmCell) { in Run()
40 const auto* src_lstm_op = static_cast<LstmCellOperator*>(src_op); in Run()
49 if (!GetMatchingRnnArray(model, src_op->outputs[kOutputTensor], in Run()
54 if (!GetMatchingRnnArray(model, src_op->outputs[kCellStateTensor], in Run()
60 int num_cell = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
63 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run()
67 model->GetArray(src_op->inputs[kRecurrentToInputWeightsTensor]) in Run()
76 src_op->outputs[kOutputTensor], src_op->outputs[kCellStateTensor])); in Run()
90 model->GetArray(src_op->inputs[kInputToInputWeightsTensor]), 0, 0); in Run()
[all …]
|
| D | propagate_activation_function_into_constants.cc | 41 auto* src_op = GetOpWithOutput(*model, ac_op->inputs[0]); in Run() local
42 if (!src_op) { in Run()
47 if (CountTrueOutputs(*model, *src_op) > 1) { in Run()
51 LogName(*ac_op), LogName(*src_op)); in Run()
56 switch (src_op->type) { in Run()
58 src_op_input = src_op->inputs[0]; in Run()
63 CHECK_EQ(src_op->outputs[0], ac_op->inputs[0]); in Run()
72 LogName(*ac_op), LogName(*src_op), src_op_input); in Run()
82 LogName(*ac_op), LogName(*src_op), src_op_input); in Run()
118 LogName(*src_op), src_op_input); in Run()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | utils_jni.cc | 20 void resolveOutputs(JNIEnv* env, const char* type, jlongArray src_op, in resolveOutputs() argument
23 jint len = env->GetArrayLength(src_op); in resolveOutputs()
36 jlong* op_handles = env->GetLongArrayElements(src_op, nullptr); in resolveOutputs()
48 env->ReleaseLongArrayElements(src_op, op_handles, JNI_ABORT); in resolveOutputs()
|
| D | utils_jni.h | 27 void resolveOutputs(JNIEnv* env, const char* type, jlongArray src_op,
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | quantize_training.cc | 81 const string& src_op = node->type_string(); in FindType() local
82 if (src_op == "Const" || src_op == "Variable" || src_op == "VariableV2") { in FindType()
85 } else if (src_op == "Relu") { in FindType()
89 } else if (src_op == "Relu6") { in FindType()
97 } else if (src_op == "Sigmoid") { in FindType()
102 } else if (src_op == "Tanh") { in FindType()
107 } else if (src_op == "Reshape" || src_op == "ConcatV2") { in FindType()
117 } else if (src_op == "Identity" || src_op == "MaxPool" || in FindType()
118 src_op == "AvgPool" || src_op == "MaxPool3D" || in FindType()
119 src_op == "AvgPool3D") { in FindType()
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | quantize.py | 277 def _CheckIfQuantizableOp(src_op, quantized_ops): argument
287 src_op_name = set([src_op.type])
288 if src_op in quantized_ops:
295 if len(src_op.outputs) == 1 and len(src_op.outputs[0].consumers()) == 1:
296 op_consumers = src_op.outputs[0].consumers()
298 logging.info('Skipping quant after %s', src_op.name)
301 input_ops = src_op.inputs
|
| /external/v8/src/wasm/baseline/arm64/ |
| D | liftoff-assembler-arm64.h | 204 MemOperand src_op = in Load() local
210 Ldrb(dst.gp().W(), src_op); in Load()
213 Ldrsb(dst.gp().W(), src_op); in Load()
216 Ldrsb(dst.gp().X(), src_op); in Load()
220 Ldrh(dst.gp().W(), src_op); in Load()
223 Ldrsh(dst.gp().W(), src_op); in Load()
226 Ldrsh(dst.gp().X(), src_op); in Load()
230 Ldr(dst.gp().W(), src_op); in Load()
233 Ldrsw(dst.gp().X(), src_op); in Load()
236 Ldr(dst.gp().X(), src_op); in Load()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 488 const string& src_op = input_edge->src()->type_string(); in ConstantPartialShape() local
493 } else if (src_op == "Shape") { in ConstantPartialShape()
495 } else if (src_op == "ShapeN") { in ConstantPartialShape()
497 } else if (src_op == "Pack") { in ConstantPartialShape()
513 } else if (src_op == "Concat" || src_op == "ConcatV2") { in ConstantPartialShape()
517 src_op == "Concat" ? 0 : src_context->num_inputs() - 1; in ConstantPartialShape()
535 } else if (src_op == "StridedSlice") { in ConstantPartialShape()
|
| /external/v8/src/wasm/baseline/mips/ |
| D | liftoff-assembler-mips.h | 313 MemOperand src_op = (offset_reg != no_reg) ? MemOperand(src, offset_imm) in Load() local
319 lbu(dst.gp(), src_op); in Load()
322 lbu(dst.low_gp(), src_op); in Load()
326 lb(dst.gp(), src_op); in Load()
329 lb(dst.low_gp(), src_op); in Load()
334 TurboAssembler::Ulhu(dst.gp(), src_op); in Load()
337 TurboAssembler::Ulhu(dst.low_gp(), src_op); in Load()
341 TurboAssembler::Ulh(dst.gp(), src_op); in Load()
344 TurboAssembler::Ulh(dst.low_gp(), src_op); in Load()
349 TurboAssembler::Ulw(dst.gp(), src_op); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/ia32/ |
| D | liftoff-assembler-ia32.h | 232 Operand src_op = offset_reg == no_reg in Load() local
239 movzx_b(dst.gp(), src_op); in Load()
242 movsx_b(dst.gp(), src_op); in Load()
245 movzx_b(dst.low_gp(), src_op); in Load()
249 movsx_b(dst.low_gp(), src_op); in Load()
254 movzx_w(dst.gp(), src_op); in Load()
257 movsx_w(dst.gp(), src_op); in Load()
260 movzx_w(dst.low_gp(), src_op); in Load()
264 movsx_w(dst.low_gp(), src_op); in Load()
269 mov(dst.gp(), src_op); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/mips64/ |
| D | liftoff-assembler-mips64.h | 277 MemOperand src_op(src_addr, offset_imm); in Load()
281 src_op = MemOperand(src, offset_imm); in Load()
288 lbu(dst.gp(), src_op); in Load()
292 lb(dst.gp(), src_op); in Load()
296 TurboAssembler::Ulhu(dst.gp(), src_op); in Load()
300 TurboAssembler::Ulh(dst.gp(), src_op); in Load()
303 TurboAssembler::Ulwu(dst.gp(), src_op); in Load()
307 TurboAssembler::Ulw(dst.gp(), src_op); in Load()
310 TurboAssembler::Uld(dst.gp(), src_op); in Load()
313 TurboAssembler::Ulwc1(dst.fp(), src_op, t8); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/x64/ |
| D | liftoff-assembler-x64.h | 211 Operand src_op = liftoff::GetMemOp(this, src_addr, offset_reg, offset_imm); in Load() local
216 movzxbl(dst.gp(), src_op); in Load()
219 movsxbl(dst.gp(), src_op); in Load()
222 movsxbq(dst.gp(), src_op); in Load()
226 movzxwl(dst.gp(), src_op); in Load()
229 movsxwl(dst.gp(), src_op); in Load()
232 movsxwq(dst.gp(), src_op); in Load()
236 movl(dst.gp(), src_op); in Load()
239 movsxlq(dst.gp(), src_op); in Load()
242 movq(dst.gp(), src_op); in Load()
[all …]
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_tgsi.c | 434 unsigned src_op, in lp_build_emit_fetch() argument
437 const struct tgsi_full_src_register *reg = &inst->Src[src_op]; in lp_build_emit_fetch()
439 tgsi_opcode_infer_src_type(inst->Instruction.Opcode, src_op); in lp_build_emit_fetch()
|
| D | lp_bld_tgsi.h | 657 unsigned src_op,
|
| D | lp_bld_tgsi_soa.c | 1973 unsigned src_op) in lp_build_lod_property() argument
1975 const struct tgsi_full_src_register *reg = &inst->Src[src_op]; in lp_build_lod_property()
|