Searched defs:list_shape (Results 1 – 5 of 5) sorted by relevance
53 const xla::Shape& list_shape, int64 num_elements) { in GetTensorListDynamicDims()171 xla::Shape list_shape; in Compile() local243 xla::Shape list_shape; in Compile() local307 xla::Shape list_shape; in Compile() local
270 xla::XlaBuilder* b, const xla::Shape& list_shape, in CreateZerosTensorListWithShape()306 xla::Shape list_shape; in GetInitializedTensorListForElement() local
354 const xla::Shape& list_shape = list_shape_or.ValueOrDie(); in Compile() local
507 const xla::Shape& list_shape = body_input_shape.tuple_shapes(i); in Compile() local
466 auto list_shape = rewriter.create<TF::ConcatOp>( in matchAndRewrite() local