- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for is_all_scalars (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
return ranked && ranked.getRank() == (axis + 1) && ranked.getShape()[axis] == 1; }); }; // Returns true if all binary ops operands at `operand_idx` index are scalars. auto is_all_scalars = [&](int operand_idx) -> bool { return llvm::all_of(op.getValues(), [&](Value arg) -> bool { if (exceptions.count(arg)) return true; auto operand = arg.getDefiningOp()->getOperand(operand_idx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)