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

Method RemoveDim

tensorflow/core/framework/tensor_shape_test.cc:437–445  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

435}
436
437void TensorShapeOld::RemoveDim(int d) {
438 CHECK_GE(d, 0);
439 CHECK_LT(d, dims());
440
441 // Update the number of elements and remove the dimension from the
442 // sizes.
443 dim_sizes_.erase(dim_sizes_.begin() + d);
444 recompute_dims();
445}
446
447void TensorShapeOld::recompute_dims() {
448 num_elements_ = 1;

Callers 15

AddToTensorAccumulatorFunction · 0.45
ComputeMethod · 0.45
py_seq_tensor.ccFile · 0.45
CompileMethod · 0.45
CompileMethod · 0.45
SubSliceMethod · 0.45
TESTFunction · 0.45
HandleSliceToElementFunction · 0.45
TryInsertManyMethod · 0.45
ComputeMethod · 0.45
ComputeMethod · 0.45

Calls 3

dimsFunction · 0.85
eraseMethod · 0.45
beginMethod · 0.45

Tested by

no test coverage detected