- Sort Score
- Result 10 results
- Languages All
Results 11 - 13 of 13 for broadcast_args (0.15 sec)
-
tensorflow/compiler/jit/mark_for_compilation_pass.cc
{"ApproxTopK", "BroadcastTo", "ExpandDims", "Fill", "NoOp", "Range", "Rank", "Reshape", "Shape", "ShapeN", "Size", "Squeeze", "Transpose", "ZerosLike", "OnesLike", "BiasAdd" /*PW + Broadcast*/, "BroadcastArgs", "BroadcastGradientArgs", "OneHot", "Concat", "ConcatV2", "ConcatOffset", "Const", "MirrorPad", "MirrorPadGrad", "Pack", "Pad", "PadV2", "Reverse", "ReverseV2", "ReverseSequence", "Slice", "Split",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
// maximum rank. if (max_rank <= max_bcast_rank) { return true; } // Checks if all operands are broadcasted by BroadcastTo ops with the shape // is calculated from the same BroadcastArgs op. In such case, all operands // will have the same shape. Operation* broadcast_args_pivot = nullptr; for (unsigned index : indices) { Operation* parent_broadcast_args =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); TF_DerivedOperandSizeAttr num_features = TF_DerivedOperandSizeAttr<0>; } def TF_BroadcastArgsOp : TF_Op<"BroadcastArgs", [Pure]> { let summary = "Return the shape of s0 op s1 with broadcast."; let description = [{ Given `s0` and `s1`, tensors that represent shapes, compute `r0`, the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)