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

Function GetTensorDim

tensorflow/core/util/tensor_format.h:435–440  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

433// specified 'dimension' according to 'tensor_format'.
434template <typename T>
435T GetTensorDim(gtl::ArraySlice<T> dimension_attributes,
436 TensorFormat tensor_format, char dimension) {
437 int index =
438 GetTensorDimIndex(tensor_format, dimension, dimension_attributes.size());
439 return dimension_attributes[index];
440}
441
442// Return the element from 'dimension_attribute' that corresponds to the
443// specified 'dimension' according to 'filter_tensor_format'.

Callers 15

ComputeMethod · 0.85
launchMethod · 0.85
Conv2DShapeImplFunction · 0.85
AvgPoolShapeFunction · 0.85
MaxPoolShapeFunction · 0.85
MaxPoolV2ShapeFunction · 0.85
ComputeMethod · 0.85
operator()Method · 0.85

Calls 4

GetTensorDimIndexFunction · 0.85
sizeMethod · 0.45
dim_sizesMethod · 0.45
shapeMethod · 0.45

Tested by

no test coverage detected