Searched defs:sa_shape (Results 1 – 1 of 1) sorted by relevance
452 std::vector<InputDesc>* inputs, TensorShape* sa_shape) { in AnalyzeInputs()515 const std::vector<InputDesc>& inputs, const TensorShape& sa_shape) { in ConstructScopedAllocatorNode()602 const TensorShape& sa_shape, in BuildSAConcatNode()844 TensorShape sa_shape; in Rewrite() local