Home
last modified time | relevance | path

Searched refs:IsShapeConsumerOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util.h92 bool IsShapeConsumerOp(const Node& node);
Dpartially_decluster_pass.cc61 if (IsShapeConsumerOp(*n)) { in FindNodesToDecluster()
365 if (!IsShapeConsumerOp(*n)) { in PartiallyDeclusterGraph()
Dxla_cluster_util.cc317 bool IsShapeConsumerOp(const Node& node) { in IsShapeConsumerOp() function
Dmark_for_compilation_pass.cc757 bool is_shape_consumer_op = n && IsShapeConsumerOp(*n); in RunEdgeContractionLoop()