- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IsConstantSmall (0.11 sec)
-
tensorflow/compiler/jit/clone_constants_for_better_clustering.cc
graph_->AddEdge(e->src(), e->src_output(), new_in, e->dst_input()); } } new_in->set_assigned_device_name(n->assigned_device_name()); return new_in; } namespace { absl::StatusOr<bool> IsConstantSmall(Node* n) { const TensorProto* proto = nullptr; TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "value", &proto)); int64_t total_elements = 1; for (const auto& dim : proto->tensor_shape().dim()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0)