Home
last modified time | relevance | path

Searched refs:list_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.cc116 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in IsTensorListInitialized()
117 *is_initialized = list_shape.IsTuple(); in IsTensorListInitialized()
127 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in IsNestedTensorList()
128 *is_nested_list = (xla::ShapeUtil::TupleElementCount(list_shape) > 2); in IsNestedTensorList()
145 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in GetTensorListBufferShape()
146 *buffer_shape = xla::ShapeUtil::GetTupleElementShape(list_shape, 0); in GetTensorListBufferShape()
166 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in GetTensorListPushIndex()
167 int tuple_size = xla::ShapeUtil::TupleElementCount(list_shape); in GetTensorListPushIndex()
179 TF_ASSIGN_OR_RETURN(xla::Shape list_shape, list.builder()->GetShape(list)); in SetTensorListPushIndex()
180 int tuple_size = xla::ShapeUtil::TupleElementCount(list_shape); in SetTensorListPushIndex()
[all …]
Dtensor_list_ops.cc53 const xla::Shape& list_shape, int64 num_elements) { in GetTensorListDynamicDims() argument
171 xla::Shape list_shape; in Compile() local
174 num_element_is_dynamic, &list_shape)); in Compile()
177 ctx, element_shape, list_shape, num_elements); in Compile()
181 ctx->builder(), list_shape, in Compile()
243 xla::Shape list_shape; in Compile() local
246 num_element_is_dynamic, &list_shape)); in Compile()
249 ctx, element_shape, list_shape, max_num_elements); in Compile()
254 ctx->builder(), list_shape, in Compile()
307 xla::Shape list_shape; in Compile() local
[all …]
Dshape_op.cc354 const xla::Shape& list_shape = list_shape_or.ValueOrDie(); in Compile() local
356 list_dynamic_dims.reserve(list_shape.tuple_shapes_size() - 1); in Compile()
357 for (int64 i = 0; i < list_shape.tuple_shapes_size() - 1; ++i) { in Compile()
360 const xla::Shape& shape = list_shape.tuple_shapes(i); in Compile()
369 ctx, CreateZerosTensorListWithShape(ctx->builder(), list_shape, in Compile()
Dwhile_op.cc507 const xla::Shape& list_shape = body_input_shape.tuple_shapes(i); in Compile() local
512 if (input_shape != list_shape) { in Compile()
515 for (int64 i = 0; i < list_shape.tuple_shapes_size() - 1; ++i) { in Compile()
518 const xla::Shape& shape = list_shape.tuple_shapes(i); in Compile()
546 ctx, CreateZerosTensorListWithShape(ctx->builder(), list_shape, in Compile()
Dtensor_list_utils.h83 xla::XlaBuilder* b, const xla::Shape& list_shape,
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc466 auto list_shape = rewriter.create<TF::ConcatOp>( in matchAndRewrite() local
476 rewriter.replaceOpWithNewOp<TF::FillOp>(op, result_type, list_shape, zero); in matchAndRewrite()