Searched refs:sa_shape (Results 1 – 1 of 1) sorted by relevance
284 std::vector<InputDesc>* inputs, TensorShape* sa_shape) { in AnalyzeInputs() argument307 *sa_shape = TensorShape({num_elts}); in AnalyzeInputs()318 const std::vector<InputDesc>& inputs, const TensorShape& sa_shape) { in ConstructScopedAllocatorNode() argument326 sa_builder.Attr("shape", sa_shape); in ConstructScopedAllocatorNode()355 const TensorShape& sa_shape, in BuildSAConcatNode() argument388 << sa_shape.DebugString(); in BuildSAConcatNode()393 sac_builder.Attr("shape", sa_shape); in BuildSAConcatNode()591 TensorShape sa_shape; in Rewrite() local596 &inputs, &sa_shape)); in Rewrite()603 input_shapes, inputs, sa_shape)); in Rewrite()[all …]