Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for kShapeInvariantAttr (0.2 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/transforms/drop_while_shape_invariant.cc 

    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h"
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"

namespace mlir {
namespace TF {

namespace {

constexpr char kShapeInvariantAttr[] = "shape_invariant";

#define GEN_PASS_DEF_DROPWHILESHAPEINVARIANTPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"

class DropWhileShapeInvariantPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 2.4K bytes
    - Viewed (0)
Back to top