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

Class Index

tensorflow/core/kernels/conv_2d_gpu.h:143–150  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

141// An index type with compile-time known size.
142template <int IndexCount>
143struct Index : Array<int, IndexCount, 0> {
144 typedef Array<int, IndexCount, 0> Base;
145 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index() : Base() {}
146 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0) : Base(a0) {}
147 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0, int a1) : Base(a0, a1) {}
148 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index(int a0, int a1, int a2)
149 : Base(a0, a1, a2) {}
150};
151
152// A helper function that converts a tensor index into a flat array index.
153template <int IndexCount>

Callers 5

ReduceMeanAcrossImageMethod · 0.70
operator()Method · 0.70
AddMethod · 0.50

Calls

no outgoing calls

Tested by

no test coverage detected