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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
convert_pure_conv_to_depthwise.cc 36 auto& weights_array = model->GetArray(conv_op->inputs[1]); local
37 if (!weights_array.buffer) {
41 if (weights_array.data_type != ArrayDataType::kFloat) {
44 if (weights_array.shape().dims(3) != 1) {
70 depthwiseconv_op->depth_multiplier = weights_array.shape().dims(0);
78 const auto& weights_shape = weights_array.shape();
80 weights_array.GetMutableBuffer<ArrayDataType::kFloat>();
93 *weights_array.mutable_shape()->mutable_dims() = {1, width, height, depth};
create_im2col_arrays.cc 38 const auto& weights_array = model->GetArray(conv_op->inputs[1]); local
39 if (!weights_array.has_shape()) {
45 const auto& weights_shape = weights_array.shape();
propagate_fixed_sizes.cc 120 const auto& weights_array = model->GetArray(weights_name);
122 if (!weights_array.has_shape()) {
156 const auto& weights_array = model->GetArray(op->inputs[1]); local
158 if (!weights_array.has_shape()) {
161 const auto& weights_shape = weights_array.shape();
198 const auto& weights_array = model->GetArray(op->inputs[1]); local
200 if (!weights_array.has_shape()) {
203 const auto& weights_shape = weights_array.shape();
317 const auto& weights_array = model->GetArray(op->inputs[1]); local
319 if (!weights_array.has_shape())
790 const auto& weights_array = local
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 345 const auto& weights_array = model.GetArray(weights_array_name); local
346 CHECK(weights_array.buffer->type == ArrayDataType::kFloat);
1241 const auto& weights_array = model.GetArray(weights_name); local
    [all...]
tooling_util.cc 1486 const auto& weights_array = model.GetArray(op->inputs[1]); local
    [all...]

Completed in 52 milliseconds