Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc239 TF::ReshapeOp GetReshapeOpForConv2DFilter(ArrayRef<int64_t> new_shape, in GetReshapeOpForConv2DFilter() function
297 GetReshapeOpForConv2DFilter(new_shape, before_reshape_value, &builder); in HandleConv2DFilter()
307 GetReshapeOpForConv2DFilter(final_shape, transpose_op, &builder); in HandleConv2DFilter()
387 GetReshapeOpForConv2DFilter(first_reshape_shape, new_backprop, &builder); in HandleConv2DBackPropFilter()
397 GetReshapeOpForConv2DFilter(last_reshape_shape, transpose_op, &builder); in HandleConv2DBackPropFilter()