Home
last modified time | relevance | path

Searched defs:list_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc53 const xla::Shape& list_shape, int64 num_elements) { in GetTensorListDynamicDims()
171 xla::Shape list_shape; in Compile() local
243 xla::Shape list_shape; in Compile() local
307 xla::Shape list_shape; in Compile() local
Dtensor_list_utils.cc270 xla::XlaBuilder* b, const xla::Shape& list_shape, in CreateZerosTensorListWithShape()
306 xla::Shape list_shape; in GetInitializedTensorListForElement() local
Dshape_op.cc354 const xla::Shape& list_shape = list_shape_or.ValueOrDie(); in Compile() local
Dwhile_op.cc507 const xla::Shape& list_shape = body_input_shape.tuple_shapes(i); in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc466 auto list_shape = rewriter.create<TF::ConcatOp>( in matchAndRewrite() local