Home
last modified time | relevance | path

Searched refs:UpdateDynamicDimension (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h407 static void UpdateDynamicDimension(Shape* shape, ShapeIndexView index,
Dshape_util.cc402 /* static */ void ShapeUtil::UpdateDynamicDimension(Shape* shape, in UpdateDynamicDimension() function in xla::ShapeUtil
412 UpdateDynamicDimension(shape->mutable_tuple_shapes(index.front()), in UpdateDynamicDimension()
Dshape_util_test.cc739 ShapeUtil::UpdateDynamicDimension(&tuple_shape, {0}, 1, true); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc1801 ShapeUtil::UpdateDynamicDimension(parameter->mutable_shape(), in Run()