HomeSort by relevance Sort by last modified time
    Searched full:weights_output (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/tensorrt/convert/
convert_nodes.cc 607 TRT_ShapedWeights weights_output = ctx.get_temp_weights_like(weights_input);
621 auto ret = UnaryCompute(weights_input, &weights_output, unary_op);
624 outputs->push_back(TRT_TensorOrWeights(weights_output));
679 TRT_ShapedWeights weights_output = ctx.get_temp_weights(dtype, output_shape);
693 auto ret = BinaryCompute(weights_input_l, weights_input_r, &weights_output,
698 outputs->push_back(TRT_TensorOrWeights(weights_output));
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 1237 const string weights_output = base + "weights"; local
    [all...]

Completed in 55 milliseconds