| /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/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.td | 36 (TF_AssignAddVariableOp:$src_op $resource, $value),
40 (CreateTFReadVariableOp $src_op, $value, $resource),
48 (TF_AssignSubVariableOp:$src_op $resource, $value),
52 (CreateTFReadVariableOp $src_op, $value, $resource),
62 (TF_ResourceApplyGradientDescentOp:$src_op $resource, $alpha, $delta, BoolAttr:$_),
66 (CreateTFReadVariableOp $src_op, $alpha, $resource),
78 (TF_ResourceApplyKerasMomentumOp:$src_op
84 (CreateTFReadVariableOp $src_op, $grad, $accum_resource),
99 (TF_ResourceApplyKerasMomentumOp:$src_op
105 (CreateTFReadVariableOp $src_op, $grad, $accum_resource),
[all …]
|
| D | lower_tf.td | 100 (TF_SparseSoftmaxCrossEntropyWithLogitsOp:$src_op
227 Pat<(TF_ZerosLikeOp:$src_op TensorOf<[AnyInteger, AnyFloat]>:$input),
229 (CreateTFShapeOp $src_op, $input, /*use 32bit*/ConstBoolAttrFalse))>;
|
| /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/core/common_runtime/ |
| D | shape_refiner.cc | 465 const string& src_op = input_edge->src()->type_string(); in ConstantPartialShape() local
470 } else if (src_op == "Cast") { in ConstantPartialShape()
504 } else if (src_op == "Shape") { in ConstantPartialShape()
506 } else if (src_op == "ShapeN") { in ConstantPartialShape()
508 } else if (src_op == "Pack") { in ConstantPartialShape()
524 } else if (src_op == "Concat" || src_op == "ConcatV2") { in ConstantPartialShape()
528 src_op == "Concat" ? 0 : src_context->num_inputs() - 1; in ConstantPartialShape()
546 } else if (src_op == "StridedSlice") { in ConstantPartialShape()
549 } else if (src_op == "VariableShape") { in ConstantPartialShape()
|
| /external/v8/src/wasm/baseline/arm64/ |
| D | liftoff-assembler-arm64.h | 346 MemOperand src_op = in LoadTaggedPointer() local
348 LoadTaggedPointerField(dst, src_op); in LoadTaggedPointer()
381 MemOperand src_op = in Load() local
387 Ldrb(dst.gp().W(), src_op); in Load()
390 Ldrsb(dst.gp().W(), src_op); in Load()
393 Ldrsb(dst.gp().X(), src_op); in Load()
397 Ldrh(dst.gp().W(), src_op); in Load()
400 Ldrsh(dst.gp().W(), src_op); in Load()
403 Ldrsh(dst.gp().X(), src_op); in Load()
407 Ldr(dst.gp().W(), src_op); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/ia32/ |
| D | liftoff-assembler-ia32.h | 333 Operand src_op = offset_reg == no_reg in Load() local
340 movzx_b(dst.gp(), src_op); in Load()
343 movsx_b(dst.gp(), src_op); in Load()
346 movzx_b(dst.low_gp(), src_op); in Load()
350 movsx_b(dst.low_gp(), src_op); in Load()
354 movzx_w(dst.gp(), src_op); in Load()
357 movsx_w(dst.gp(), src_op); in Load()
360 movzx_w(dst.low_gp(), src_op); in Load()
364 movsx_w(dst.low_gp(), src_op); in Load()
368 mov(dst.gp(), src_op); in Load()
[all …]
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_tgsi.c | 443 unsigned src_op, in lp_build_emit_fetch() argument
446 const struct tgsi_full_src_register *reg = &inst->Src[src_op]; in lp_build_emit_fetch()
448 tgsi_opcode_infer_src_type(inst->Instruction.Opcode, src_op); in lp_build_emit_fetch()
|
| D | lp_bld_tgsi.h | 742 unsigned src_op,
|
| D | lp_bld_tgsi_soa.c | 2010 unsigned src_op) in lp_build_lod_property() argument
2012 const struct tgsi_full_src_register *reg = &inst->Src[src_op]; in lp_build_lod_property()
|
| /external/v8/src/wasm/baseline/mips/ |
| D | liftoff-assembler-mips.h | 410 MemOperand src_op = (offset_reg != no_reg) ? MemOperand(src, offset_imm) in Load() local
416 lbu(dst.gp(), src_op); in Load()
419 lbu(dst.low_gp(), src_op); in Load()
423 lb(dst.gp(), src_op); in Load()
426 lb(dst.low_gp(), src_op); in Load()
431 TurboAssembler::Ulhu(dst.gp(), src_op); in Load()
434 TurboAssembler::Ulhu(dst.low_gp(), src_op); in Load()
438 TurboAssembler::Ulh(dst.gp(), src_op); in Load()
441 TurboAssembler::Ulh(dst.low_gp(), src_op); in Load()
446 TurboAssembler::Ulw(dst.gp(), src_op); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/x64/ |
| D | liftoff-assembler-x64.h | 272 Operand src_op = liftoff::GetMemOp(this, src_addr, offset_reg, in LoadTaggedPointer() local
274 LoadTaggedPointerField(dst, src_op); in LoadTaggedPointer()
320 Operand src_op = liftoff::GetMemOp(this, src_addr, offset_reg, offset_imm); in Load() local
325 movzxbl(dst.gp(), src_op); in Load()
328 movsxbl(dst.gp(), src_op); in Load()
331 movsxbq(dst.gp(), src_op); in Load()
335 movzxwl(dst.gp(), src_op); in Load()
338 movsxwl(dst.gp(), src_op); in Load()
341 movsxwq(dst.gp(), src_op); in Load()
345 movl(dst.gp(), src_op); in Load()
[all …]
|
| /external/v8/src/wasm/baseline/mips64/ |
| D | liftoff-assembler-mips64.h | 409 MemOperand src_op = liftoff::GetMemOp(this, src_addr, offset_reg, offset_imm); in Load() local
415 Lbu(dst.gp(), src_op); in Load()
419 Lb(dst.gp(), src_op); in Load()
423 TurboAssembler::Ulhu(dst.gp(), src_op); in Load()
427 TurboAssembler::Ulh(dst.gp(), src_op); in Load()
430 TurboAssembler::Ulwu(dst.gp(), src_op); in Load()
434 TurboAssembler::Ulw(dst.gp(), src_op); in Load()
437 TurboAssembler::Uld(dst.gp(), src_op); in Load()
440 TurboAssembler::Ulwc1(dst.fp(), src_op, t8); in Load()
443 TurboAssembler::Uldc1(dst.fp(), src_op, t8); in Load()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_patterns.td | 159 def : Pat<(TF_SelectV2Op:$src_op $cond, $x, $y), (TFL_SelectOp $cond, $x, $y), [(HasSameStaticShape…
160 def : Pat<(TF_SelectV2Op:$src_op $cond, $x, $y), (TFL_SelectV2Op $cond, $x, $y), [(HasNotSameStatic…
|
| /external/v8/src/wasm/baseline/arm/ |
| D | liftoff-assembler-arm.h | 590 MemOperand src_op = variable
595 __ ldrb(dst.gp(), src_op);
598 __ ldrb(dst.low_gp(), src_op);
602 __ ldrsb(dst.gp(), src_op);
605 __ ldrsb(dst.low_gp(), src_op);
609 __ ldrh(dst.gp(), src_op);
612 __ ldrh(dst.low_gp(), src_op);
616 __ ldrsh(dst.gp(), src_op);
619 __ ldr(dst.gp(), src_op);
622 __ ldrsh(dst.low_gp(), src_op);
[all …]
|