MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / RemoveLastDims

Method RemoveLastDims

tensorflow/core/framework/tensor_shape.h:235–238  ·  view source on GitHub ↗

\brief Removes last `n` dimensions from the `TensorShape`. REQUIRES: `0 <= n <= dims()`

Source from the content-addressed store, hash-verified

233 /// \brief Removes last `n` dimensions from the `TensorShape`.
234 /// REQUIRES: `0 <= n <= dims()`
235 void RemoveLastDims(int n) {
236 CHECK_LE(n, dims());
237 RemoveDimRange(dims() - n, dims());
238 }
239
240 /// \brief Removes the dimensions in range `[begin:end)` from `TensorShape`.
241 /// Negative values of `end` are interpreted as `dims() + end + 1` (as in

Callers 15

CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
CompileMethod · 0.80
XlaGatherFunction · 0.80
TESTFunction · 0.80
DoScatterNdFunction · 0.80
ComputeMethod · 0.80
ComputeAsyncMethod · 0.80
ComputeMethod · 0.80
ComputeAsyncMethod · 0.80

Calls 1

dimsFunction · 0.85

Tested by 1

TESTFunction · 0.64