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