Searched defs:list_shape (Results 1 – 5 of 5) sorted by relevance
54 const xla::Shape& list_shape, int64_t num_elements) { in GetTensorListDynamicDims()175 xla::Shape list_shape; in Compile() local249 xla::Shape list_shape; in Compile() local313 xla::Shape list_shape; in Compile() local
272 xla::XlaBuilder* b, const xla::Shape& list_shape, in CreateZerosTensorListWithShape()308 xla::Shape list_shape; in GetInitializedTensorListForElement() local
388 const xla::Shape& list_shape = list_shape_or.ValueOrDie(); in Compile() local
520 const xla::Shape& list_shape = body_input_shape.tuple_shapes(i); in Compile() local
610 auto list_shape = rewriter.create<TF::ConcatOp>( in matchAndRewrite() local