Home
last modified time | relevance | path

Searched defs:transform_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_space_to_depth_pass.cc134 SmallVector<int64_t, 4> transform_shape = { in HandlePad() local
165 SmallVector<int64_t, 4> transform_shape = { in HandleConv2DInput() local
384 SmallVector<int64_t, 4> transform_shape = { in BuildSpaceToDepth() local
/aosp_15_r20/external/pytorch/torch/export/
H A Ddynamic_shapes.py907 def transform_shape(path, t, dynamic_shape): function