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

  /external/tensorflow/tensorflow/tools/graph_transforms/
fold_batch_norms.cc 87 Tensor scaled_weights(DT_FLOAT, weights.shape());
88 auto scaled_weights_matrix = scaled_weights.flat_inner_dims<float>();
101 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node);
fold_old_batch_norms.cc 124 Tensor scaled_weights(DT_FLOAT, weights.shape());
125 auto scaled_weights_matrix = scaled_weights.flat_inner_dims<float>();
144 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node);
  /external/tensorflow/tensorflow/core/grappler/optimizers/
arithmetic_optimizer.cc 785 // Create new node `scaled_weights`.
786 NodeDef* scaled_weights = AddNode( local
788 scaled_weights->set_op("Mul");
789 scaled_weights->set_device(weights->device());
790 (*scaled_weights->mutable_attr())["T"] =
792 nodes_to_simplify->PushBack(scaled_weights);
795 scaled_weights->add_input(conv->input(1));
796 node_map_->AddOutput(weights->name(), scaled_weights->name());
797 scaled_weights->add_input(mul->input(1));
798 node_map_->AddOutput(scale->name(), scaled_weights->name())
    [all...]
  /external/tensorflow/tensorflow/python/tools/
optimize_for_inference_lib.py 325 scaled_weights = np.copy(weights)
327 scaled_weights, flags=["multi_index"], op_flags=["readwrite"])
338 scaled_weights, weights.dtype.type, weights.shape)))

Completed in 72 milliseconds