Searched refs:GetKernelSize (Results 1 – 3 of 3) sorted by relevance
557 struct GetKernelSize {563 struct GetKernelSize<Dynamic> {581 const int num_x_input = last_x - first_x + GetKernelSize<StaticKernelSize>()(kernelSize);607 for (int k = 0; k < GetKernelSize<StaticKernelSize>()(kernelSize); ++k) {630 const int num_x_input = last_x - first_x + GetKernelSize<StaticKernelSizeX>()(kernelSizeX);635 const int num_y_input = last_y - first_y + GetKernelSize<StaticKernelSizeY>()(kernelSizeY);668 for (int l = 0; l < GetKernelSize<StaticKernelSizeY>()(kernelSizeY); ++l) {672 for (int k = 0; k < GetKernelSize<StaticKernelSizeX>()(kernelSizeX); ++k) {
75 xla::StatusOr<std::vector<int64>> GetKernelSize(XlaOpKernelContext* ctx) { in GetKernelSize() function in tensorflow::__anon3074d43f0111::PoolingOp162 auto ksize_or_error = GetKernelSize(ctx); in Compile()215 auto ksize_or_error = GetKernelSize(ctx); in Compile()
127 std::vector<int64> GetKernelSize(const OpInfo& op_info) { in GetKernelSize() function1466 std::vector<int64> ksize = GetKernelSize(op_info); in OpDimensionsFromInputs()