Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc407 ShapeUtil::AppendMajorDimension(group_size, &reshaped_activation_shape); in HandleConvolution()
418 ShapeUtil::AppendMajorDimension(1, &reshaped_filter_shape); in HandleConvolution()
424 ShapeUtil::AppendMajorDimension(1, &new_output_shape); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc141 ShapeUtil::AppendMajorDimension(1, &index_shape); in TorchGather()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h355 static void AppendMajorDimension(int bound, Shape* shape);
Dshape_util.cc268 /* static */ void ShapeUtil::AppendMajorDimension(int bound, Shape* shape) { in AppendMajorDimension() function in xla::ShapeUtil