- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetDummyParams (0.16 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
void runOnOperation() override; }; // Returns true iff the broadcast val can be substituted with an XlaAllReduce. // Sets `zero` and `shape` as params for the dummy zeros that will be created. bool GetDummyParams(OpBuilder& builder, Value val_bcast, Attribute& zero, DenseIntElementsAttr& shape) { Type type = val_bcast.getType(); Type elem_type = getElementTypeOrSelf(type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)