Home
last modified time | relevance | path

Searched refs:MakeStaticShape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc115 Shape buffer_shape_static = ShapeUtil::MakeStaticShape(buffer_shape); in ReadDynamicShapes()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_xfeed.cc296 Shape buffer_shape_static = ShapeUtil::MakeStaticShape(buffer_shape); in ReadDynamicShapesOnCpu()
Dir_emitter.cc2274 ShapeUtil::ByteSizeOf(ShapeUtil::MakeStaticShape(hlo->shape())); in HandleSliceToDynamic()
2332 ShapeUtil::ByteSizeOf(ShapeUtil::MakeStaticShape(input_shape)); in HandlePadToStatic()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc222 Shape buffer_shape_static = ShapeUtil::MakeStaticShape(buffer_shape); in ReadDynamicShapes()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h270 static Shape MakeStaticShape(const Shape& original);
Dshape_util.cc396 /* static */ Shape ShapeUtil::MakeStaticShape(const Shape& original) { in MakeStaticShape() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc165 xla::ShapeUtil::MakeStaticShape(compile_time_shape); in UpdateMetadata()
/external/tensorflow/tensorflow/compiler/xla/client/
Dvalue_inference.cc1093 ShapeUtil::MakeStaticShape(Shape(root->shape()))); in AnalyzeIsDynamic()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc488 ? xla::ShapeUtil::MakeStaticShape(instruction_shape) in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1447 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView()