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

Method dim_sizes

tensorflow/core/framework/tensor_shape.cc:489–495  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

487
488template <class Shape>
489gtl::InlinedVector<int64, 4> TensorShapeBase<Shape>::dim_sizes() const {
490 gtl::InlinedVector<int64, 4> result;
491 for (auto dim : *this) {
492 result.push_back(dim.size);
493 }
494 return result;
495}
496
497template <class Shape>
498void TensorShapeBase<Shape>::set_dim(int d, int64 size) {

Callers 15

do_periodic_resample_opFunction · 0.45
TFE_Py_EncodeTensorFunction · 0.45
RunMethod · 0.45
matchAndRewriteMethod · 0.45
matchAndRewriteMethod · 0.45
DimensionSizesMethod · 0.45
SetZeroValueMethod · 0.45
ConstantInputMethod · 0.45
ReadVariableInputTensorFunction · 0.45
OneHotMethod · 0.45
CompileMethod · 0.45

Calls 1

push_backMethod · 0.45

Tested by 6

TEST_FFunction · 0.36
TestAddShapeFunction · 0.36
TestMulShapeFunction · 0.36
MatMulBCastToStrFunction · 0.36