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