Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for kXlaInferredShapesAttrName (0.22 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/encapsulate_util.h 

    // Attribute marking output tensor shapes inferred by XLA. Attribute value is
// a list of PartialTensorShape objects.
extern const char kXlaInferredShapesAttrName[];

// Infers output shapes for all nodes in graph `g`. The output shapes will be
// stored in node attribute `kXlaInferredShapesAttrName`.
//
// We have to perform shape inference before encapsulation because after
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 7.4K bytes
    - Viewed (0)
Back to top